ubuntu18.04配置CUDA-11.3、cuDNN、PyTorch

ubuntu18.04配置CUDA-11.3、cuDNN、PyTorch

0 准备工作

硬件：RTX 3070, ubuntu18.04

0.1 查看显卡驱动

查看显卡驱动各版本以及支持的最大CUDA版本,显卡驱动版本为495.46，最大支持CUDA版本为11.5

0.2 查看GPU驱动与CUDA对应关系

官网：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
表2. CUDA工具包和CUDA小版本兼容性所需的最低驱动版本

表3. CUDA工具包和对应的驱动程序版本

0.3 CUDA与PyTorch的版本对应关系

参考：https://www.cnblogs.com/zhoujiayingvana/p/15827369.html

0.4 选择CUDA版本

结合0.2和0.3 选择CUDA版本为11.3

0.5 下载CUDA

进入英伟达的cuda版本列表：https://developer.nvidia.com/cuda-toolkit-archive

进入11.3版本的列表，依次选择箭头位置,下方会出现下载与安装命令

运行下载命令进行下载：wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run
注：我这里使用命令将其下载到我的T7外接移动硬盘时报了个错误，导致下载进度没在99%处丢失(不知道是不是外接移动硬盘的问题):

我这里直接点击链接，在浏览器下载；使用手机流量可能会比校园网快很多

0.6 下载cuDNN

进入网页进行下载，需要注册或登录
登录进去后，选择Download cuDNN v8.6.0 (October 3rd, 2022), for CUDA 11.x(0.4步选择安装CUDA 11.3)

接着选Local Installer for Linux x86_64 (Tar),点击即可下载

1 安装CUDA和cuDNN

1.1 安装CUDA

给.run文件以可执行权限：chmod +x cuda_11.3.0_465.19.01_linux.run
运行安装命令：sudo sh cuda_11.3.0_465.19.01_linux.run
(注：这个命令后可能需要等一会)
选continue：

其他安装步骤依次为：accept、使用enter勾选掉Driver、选择install
我之前写的流程此处比较详细：https://blog.csdn.net/BIT_HXZ/article/details/123374962 的6.3.2 安装及选项

如果弹出A symlink already exists at /usr/local/cuda. Update to this installation? ,说明以前安装过cuda，询问你本次安装是否需要替换掉之前安装cuda产生的链接。之前安装的cuda是11.0，不太想用了，于是我这里选yes（注：yes后也需要等待安装）

安装成功，位置如下：

1.2 测试CUDA

cd /usr/local/cuda/samples/1_Utilities/deviceQuery #由自己电脑目录决定
sudo make
sudo ./deviceQuery
1
2
3

1.3 配置cuda到~/.bashrc

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.3
export PATH=/usr/local/cuda-11.3/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH
1
2
3

1.4 配置cuDNN

解压下载的文件并进入：

参考官网指导将对应文复制后赋予权限：

我这里将文件复制到自己安装的目录/usr/local/cuda-11.3中，而不是官网参考的/usr/local/cuda,这是因为上面执行了1.3 配置cuda到~/.bashrc步骤

sudo cp ./include/cudnn*.h /usr/local/cuda-11.3/include
sudo cp ./lib/libcudnn* /usr/local/cuda-11.3/lib64
 
sudo chmod a+r /usr/local/cuda-11.3/include/cudnn*.h 
sudo chmod a+r /usr/local/cuda-11.3/lib64/libcudnn*
1
2
3
4
5

检查是否安装成功：nvcc -V

2 安装pytorch1.12.0

进入pytorch官网页面：https://pytorch.org/get-started/previous-versions/
选一个不是很新，也不是很旧的版本，这里选了1.12.0

在python环境中输入：conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch

成功运行网络：

3 参考链接

非root用户安装cuda与cudnn
ubuntu系统(八)：ubuntu18.04双系统安装+ros安装+各种软件安装+深度学习环境配置全家桶
PyTorch和CUDA版本对应关系
Ubuntu18.04安装cuda11.3