Deepin20.2.4安装cuda和cudnn

Deepin20.2.4安装cuda和cudnn

前提条件:安装好Nvidia驱动 参照前文Deepin20.2.4安装Nvidia驱动教程

一、下载安装CUDA

1.1下载CUDA

NVIDIA官网下载CUDA
前文我下载的驱动通过nvidia-smi命令查看发现cuda为11.5,但是这里我们尽量选择低版本,我试了11.5会报错,因此这里我们选择11.1版本的cuda。
Deepin20.2.4安装cuda和cudnn_第1张图片
按照如下的选项进行选择:Deepin20.2.4安装cuda和cudnn_第2张图片
此时可以在命令行输入如下命令来下载,或者复制https链接到浏览器来下载也可。

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run

1.2安装CUDA

进入到cuda下载目录执行如下操作:

chmod +x cuda_11.1.0_455.23.05_linux.run #设置执行权限
sudo ./cuda_11.1.0_455.23.05_linux.run

执行时间有点长等待一段时间后可看到下图:
Deepin20.2.4安装cuda和cudnn_第3张图片
这里我们输入accept后按回车进入如下界面:
Deepin20.2.4安装cuda和cudnn_第4张图片
这个界面要注意,因为我们之前已经安装过显卡驱动了,因此此处应该取消勾选Driver,方法是使用上下方向键选择Driver,然后回车即可,最后选到Install,进行安装。
安装成功会输出“Summary”。
Deepin20.2.4安装cuda和cudnn_第5张图片

1.3添加环境变量

命令行输入

vim ~/.bashrc


进入编辑模式,在最后加入如下代码将安装目录加入到环境变量中:

# 添加统一的Cuda路径,以防万一为了方便维护cuda多个版本
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64

按esc后输入wq保存。
之后输入如下命令来刷新环境变量:

source ~/.bashrc

打开终端,输入以下命令在系统的/usr/local/下创建toolkit11.1的链接

sudo ln -s /usr/local/cuda-11.1/ /usr/local/cuda

先打开终端验证:

cd ~/NVIDIA_CUDA-11.1_Samples/1_Utilities/deviceQuery
make
./deviceQuery

期间可能会报错,g++的问题,安装g++:
sudo apt-get install g++
重新输入上述代码出现pass即为成功:Deepin20.2.4安装cuda和cudnn_第6张图片

此时在终端输入nvcc -V来验证查看当前 cuda 版本。

nvcc -V

显示如下内容则成功:
Deepin20.2.4安装cuda和cudnn_第7张图片

二、下载CUDNN

2.1下载cudnn

官网下载cudnn
这里是需要注册账号登录后才可以下载的。
这里选择好版本(需要跟CUDA11.1相符合),并选择for linux版本。如图所示:
Deepin20.2.4安装cuda和cudnn_第8张图片

2.2安装cudnn

cudnn的安装其实就是复制一些库文件和头文件到cuda中,找到cudnn的下载目录(一般都是在“下载”里面)进行解压缩,然后复制文件即可:
在cudnn的目录输入如下命令:

tar -zxvf cudnn-11.1-linux-x64-v8.0.5.39.tgz

之后,进入解压后的cuda目录,复制两个文件夹到CUDA的安装目录下的Inlude和lib64下(注意:此时的位置仍然在“下载”里面打开命令行,如果已经进入cuda中则下面前两行最前方的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*

接着查看cudnn版本:

cat /usr/local/cuda-11.1/include/cudnn.h

显示如下:
Deepin20.2.4安装cuda和cudnn_第9张图片

你可能感兴趣的:(环境搭建,python,深度学习)