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。
按照如下的选项进行选择:
此时可以在命令行输入如下命令来下载,或者复制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
执行时间有点长等待一段时间后可看到下图:
这里我们输入accept后按回车进入如下界面:
这个界面要注意,因为我们之前已经安装过显卡驱动了,因此此处应该取消勾选Driver,方法是使用上下方向键选择Driver,然后回车即可,最后选到Install,进行安装。
安装成功会输出“Summary”。
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即为成功:
此时在终端输入nvcc -V来验证查看当前 cuda 版本。
nvcc -V
显示如下内容则成功:
二、下载CUDNN
2.1下载cudnn
官网下载cudnn 这里是需要注册账号登录后才可以下载的。
这里选择好版本(需要跟CUDA11.1相符合),并选择for linux版本。如图所示:
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
显示如下: