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.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.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*
检查是否安装成功: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