【ubuntu】安装cuda cudnn

安装cuda

nvidia官网选择cuda版本:https://developer.nvidia.com/cuda-toolkit-archive

这里根据自己的需要选着版本，但是不要高于系统推荐版本。笔者这里选用11.1.0的

这里我们选择runfile(local)的安装方式

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

在运行sudo sh cuda_11.1.0_455.23.05_linux.run时，不要安装驱动、不要安装驱动、不要安装驱动，重要的事情说三遍，操作为：选择continue->输入accept->按空格取消Driver选项->选择Install安装。

安装完成后需要配置环境变量：

vim ~/.bashrc

在末尾添加：

export PATH=$PATH:/usr/local/cuda/bin  
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64  
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

最后，

source ~/.bashrc

使之生效。

安装cudnn

官网下载cuda对应版本的cudnn：https://developer.nvidia.com/rdp/cudnn-archive#a-collapse805-111

wget 和 curl 方式下载一下三个文件：
https://developer.download.nvidia.cn/compute/redist/cudnn/v8.2.1/Ubuntu18_04-x64/

wget  https://developer.download.nvidia.cn/compute/redist/cudnn/v8.2.1/Ubuntu18_04-x64/libcudnn8-dev_8.2.1.32-1+cuda11.3_amd64.deb
wget https://developer.download.nvidia.cn/compute/redist/cudnn/v8.2.1/Ubuntu18_04-x64/libcudnn8-samples_8.2.1.32-1+cuda11.3_amd64.deb
wget https://developer.download.nvidia.cn/compute/redist/cudnn/v8.2.1/Ubuntu18_04-x64/libcudnn8_8.2.1.32-1+cuda11.3_amd64.deb

安装

sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.1_amd64.deb

查看cudnn版本：

dpkg -l | grep cudnn