unbuntu20.04安装cuda11.2 + cudnn8.1.1以及多版本cuda切换

安装Cuda11.2

1. 下载安装包

wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run

2. 执行安装命令

sudo sh cuda_11.2.0_460.27.04_linux.run

输入命令后，会出现如下的界面，这个时候选择continue

这里输入accept

将光标移动到Driver选项并回车，取消选中后install：

安装完成后，显示如下：

最后不要忘记配置cuda的环境变量

vim ~/.bashrc
#按o进入编辑模式，并在最后一行添加如下代码
export PATH=$PATH:/usr/local/cuda-11.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.0/lib64
#esc后输入wq退出保存

生效更改

source ~/.bashrc

到此安装完成，输入 nvcc -V 验证 显示：

安装cudnn8.1.1

1. 首先进入nvidia官网，如下图所示，选择对应的安装包进行下载
   
2. 解压cudnn
   根据官网，和CUDA11.1配套的是CUDNN8.1.1，下载下来的是一个tar文件，那么首先需要解压这个文件，解压之后得到一个cuda文件夹，里面有lib64、include还有一个不知道什么用的文档

tar -xvf cudnn-11.2-linux-x64-v8.1.1.33.tgz

1. 将解压文件放到指定位置
   将解压缩的文件放入对应cuda项目下，并设上只读权限。

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

4.验证
查看cudnn版本信息：

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

如下图所示，可以看到已经安装成功cudnn8.1.1版本

多版本cuda切换

1. 同样的，进入nvidia官网 https://developer.nvidia.com/cuda-toolkit-archive
   选择对应的cuda版本进行下载，这里选择cuda11.1进行下载

2. 按空格根据需要选择要安装的模块，有“X”的表示安装，没有的表示不安装，我这里已经安装过418.116的显卡驱动了，所以选择不安装驱动 （最终的结果和此处的图一致，如果不一致请保持一致的选择）：
   
   按上下键移动到CUDA Toolkit 11.1上，然后按“A”键，全部不安装，然后选择Done
   
   选择完之后选择Instal
   

3. 安装完成之后，我们可以通过如下命令，来查看自己的机器下到底有多少个cuda版本，通常不带版本号的cuda是其他带版本号的cuda-x.x的软连接。

   ls -l /usr/local | grep cuda
   

   如下所示，当前cuda指向的是11.2版本
   注意此时如果nvcc -V的输出还是更改之前的CUDA版本的话，要修改环境变量：

   export PATH=/usr/local/cuda/bin:$PATH
   export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
   

   并且要去 ~/.bashrc 中查看以下是不是会显式地指定CUDA版本如：

   export PATH=/usr/local/cuda-11.1/bin:$PATH
   export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH
   

   如果有这两句的话，直接换成上面两句指向软链接 /usr/local/cuda 的两句即可。

   切换cudnn

   准确的说切换 cudnn 的过程就是重新安装一遍 cudnn，因为 cudnn 的安装本身就是 copy 一些文件到 cuda 中，然后给予这些文件运行的权限。这里安装过程就不重复赘述了。