ubuntu系统安装/卸载cuda和cudnn

如果不知道安装什么版本，本文建议安装cuda11.6+cudnn8.4.1

1.cuda安装

（1）进入cuda下载界面：

cuda下载页面，选择自己的系统版本号，下载cuda（当然也可以直接把cuda下载地址复制到浏览器中下载）

（2）安装cuda

# 赋予权限
sudo chmod a+x cuda_12.0.1_525.85.12_linux.run
# 安装cuda
sudo ./cuda_12.0.1_525.85.12_linux.run

（3）配置cuda

打开终端，输入sudo gedit ~/.bashrc将下面两行复制到文件最下面，注意需要修改自己的cuda版本号

export PATH=/usr/local/cuda-12.0/bin${
    
    PATH:+:${
    
    PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

（4）验证cuda

打开终端，输入nvcc -V即可查看到版本号

2.安装cudnn

（1）下载cudnn

cudnn官网，目前cuda12.0还没有匹配的cudnn，所以建议还是安装cuda11.6及以下；建议下载tar包

（2）cudnn安装

将下载的tar文件解压到当前目录，进入该目录，打开终端，输入下述命令：

cd cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive
sudo cp include/cudnn*.h /usr/local/cuda-11.6/include/
sudo cp -P lib/libcudnn* /usr/local/cuda-11.6/lib64
sudo chmod a+r /usr/local/cuda-11.6/include/cudnn*.h /usr/local/cuda-11.6/lib64/libcudnn*

至此cuda+cudnn已经全部安装完毕

3.卸载cuda

卸载cuda很简单，只需要简单的两部即可：

（1）卸载安装包

打开终端，输入如下指令：

cd /usr/local/cuda-11.6/bin
sudo ./cuda-uninstaller

（2）删除配置

打开终端，输入sudo gedit ~/.bashrc将最下面和cuda有关的删除即可

4.卸载cudnn

此方法只针对通过deb安装的cudnn，如果通过tar包安装的cudnn，直接删除cuda-11.6文件夹即可

（1）卸载安装包

通过deb查找cudnn安装包

sudo dpkg -l|grep cudnn

卸载cudnn安装包

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.5.0.96

（2）删除配置文件

此时如果再运行一遍sudo dpkg -l|grep cudnn，会发现还有残留的配置，通过下述语句删除即可

dpkg -l | grep ^rc | cut -d' ' -f3 | sudo xargs dpkg --purge