Linux/Ubuntu环境下的CUDA升级
首先需要知道Ubuntu的驱动最高支持哪个版本的CUDA!!!(很重要)
可通过下图查看:
其中:
- 1 是Ubuntu驱动版本
- 2 是CUDA版本
具体查看命令可参考如下代码:
查看CUDA命令:
nvidia-smi
nvcc -V
查看驱动:
ubuntu-drivers devices
根据上图了解Ubuntu的版本后,可根据下图去寻找对应可支持的CUDA版本(版本一定要对应)
在了解好Linux/Ubuntu支持的CUDA版本后,在官方网站下载对应的CUDA并安装,官方下载连接: https://developer.nvidia.com/cuda-toolkit-archive
具体如下图所示:
在Base Installer下两行可执行命令,打开命令行后,执行命令(第二个需要管理员密码)
执行第一个命令:
wget https:developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
等待下载完成即可,下载好的文件会在主目录下home
执行第二个命令:
sudo sh cuda_11.8.0_520.61.05_linux.run
输入管理员密码进行安装即可,具体操作如下所示:
选择Continue
输入accept
取消勾选Driver,install
选择No,具体配置在bashrc中操作
显示下图即安装成功
这里先用nvcc -V查看一下CUDA版本,仍为10.2。这是因为还没有在bashrc中配置CUDA11.8的环境
接着在bashrc中进行配置即可
进入./(home目录找到.bashrc文件,双击点开修改即可)
(ps:这里如果有vim佬,那用vim就可以了,vim ~/.bashrc,我不会vim!!!)
对CUDA配置进行修改就行(记得保存)
最后再执行source ~/.bashrc(一定要执行)
CUDA11.8,结束!!!