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

biter0088

已于 2023-06-13 00:54:58 修改

阅读量4k

点赞数 6

分类专栏：深度学习文章标签： pytorch 深度学习人工智能 cuda cuDNN

于 2022-10-30 20:36:23 首次发布

本文链接：https://blog.csdn.net/BIT_HXZ/article/details/127604530

版权

深度学习专栏收录该内容

14 篇文章 3 订阅

订阅专栏

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

文章目录

0 准备工作
1 安装CUDA和cuDNN
2 安装pytorch1.12.0
3 参考链接

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*

在这里插入图片描述