Ubuntu20.04下更换CUDA版本

一、卸载原始CUDA

法一：控制台卸载(10.1版本以下)

• 查看版本为10.1

nvcc -V

• 依次运行如下命令

sudo apt-get remove cuda
sudo apt autoremove 
sudo apt-get remove cuda*

• 删除对应的cuda文件夹

cd /usr/local/
# sudo rm -r cuda-version
sudo rm -r cuda

• 查看并卸载剩余残留

查看

sudo dpkg -l |grep cuda

删除所有残留

sudo dpkg -P nvidia-cuda-toolkit
sudo dpkg -P nvidia-cuda-gdb
sudo dpkg -P nvidia-cuda-doc
sudo dpkg -P nvidia-cuda-dev

再次查看，发现已经没有CUDA了

法二：cuda-uninstaller(10.1版本以上)

进入/usr/local/cuda-version/bin目录，打开终端，运行

sudo ./cuda-uninstaller

选中所有选项，选择Done。

删除文件夹

sudo rm -rf /usr/local/cuda-version

二、安装新的CUDA

在链接中选择想安装的版本，这里我选择11.7：

https://developer.nvidia.com/cuda-toolkit-archive

选择完毕后，下方会弹出一个Installation Instructions，依次运行代码：

sudo wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

完成后再进行系统变量的配置（注意要修改对应版本）

echo 'export PATH="usr/local/cuda-11.7/bin:$PATH"' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH="usr/local/cuda-11.7/LIB64:$LD_LIBRARY_PATH"' >> ~/.bashrc
source ~/.bashrc
nvcc -V

运行完后发现版本更换了

三、安装cuDNN

1、安装

• 到链接下安装对应版本

https://developer.nvidia.com/rdp/cudnn-archive

• 按照官方说明，修改对应部分。依次如下命令安装

  

  其中第一句运行完后会给出第二句的口令
  

  sudo dpkg -i cudnn-local-repo-ubuntu2004-8.7.0.84_1.0-1_amd64.deb
  sudo cp /var/cudnn-local-repo-ubuntu2004-8.7.0.84/cudnn-local-A3837CDF-keyring.gpg /usr/share/keyrings/
  sudo apt-get update
  sudo apt-get install libcudnn8=8.7.0.84-1+cuda11.8
  sudo apt-get install libcudnn8-dev=8.7.0.84-1+cuda11.8
  sudo apt-get install libcudnn8-samples=8.7.0.84-1+cuda11.8
  

2、检测

• 查询

lsmod | grep nouveau

• 提前安装一些库，否则会报错

sudo apt-get install libfreeimage3 libfreeimage-dev

• 进入/usr/src/cudnn_samples_v8/mnistCUDNN目录，运行命令：

cd /usr/src/cudnn_samples_v8/mnistCUDNN
sudo make clean && sudo make

如果前面库安装了，应该就不会报错。

• 运行测试

./mnistCUDNN

最后显示测试通过

参考链接：

环境配置之cuda的卸载（ubuntu）

Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程（图文）

Ubuntu20.04安装cuDNN（包括WSL）