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

☼Brilliant

已于 2022-06-18 23:43:24 修改

阅读量2.1k

点赞数 5

文章标签：深度学习 linux ubuntu

于 2022-06-12 12:30:04 首次发布

本文链接：https://blog.csdn.net/qq_41295081/article/details/125240142

版权

本文详细介绍了如何在Linux环境下安装CUDA 11.2和CUDNN 8.1.1，包括下载安装包、执行安装命令、配置环境变量等步骤。同时，还阐述了如何验证安装成功以及如何在已有的CUDA版本之间进行切换。此外，文中还提供了多版本CUDA的管理方法，确保能灵活使用不同版本的CUDA工具包。

摘要由CSDN通过智能技术生成

安装Cuda11.2

下载安装包

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

执行安装命令

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

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

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

在这里插入图片描述

将解压文件放到指定位置
将解压缩的文件放入对应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切换

同样的，进入nvidia官网 https://developer.nvidia.com/cuda-toolkit-archive
选择对应的cuda版本进行下载，这里选择cuda11.1进行下载
按空格根据需要选择要安装的模块，有“X”的表示安装，没有的表示不安装，我这里已经安装过418.116的显卡驱动了，所以选择不安装驱动（最终的结果和此处的图一致，如果不一致请保持一致的选择）：

按上下键移动到CUDA Toolkit 11.1上，然后按“A”键，全部不安装，然后选择Done

选择完之后选择Instal
安装完成之后，我们可以通过如下命令，来查看自己的机器下到底有多少个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 中，然后给予这些文件运行的权限。这里安装过程就不重复赘述了。