unbuntu20.04安装cuda11.2 + cudnn8.1.1

安装Cuda11.2

1. 下载安装包

wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run

2. 执行安装命令

sudo sh cuda_11.2.0_460.27.04_linux.run

输入命令后，会出现如下的界面，这个时候选择continue

这里输入accept

将光标移动到Driver选项并回车，取消选中后install：

安装完成后，显示如下：

最后不要忘记配置cuda的环境变量

vim ~/.bashrc
#按o进入编辑模式，并在最后一行添加如下代码
export PATH=$PATH:/usr/local/cuda-11.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.0/lib64
#esc后输入wq退出保存

生效更改

source ~/.bashrc

到此安装完成，输入 nvcc -V 验证 显示：

安装cudnn8.1.1

1. 首先进入nvidia官网，如下图所示，选择对应的安装包进行下载
   
2. 解压cudnn
   根据官网，和CUDA11.1配套的是CUDNN8.1.1，下载下来的是一个tar文件，那么首先需要解压这个文件，解压之后得到一个cuda文件夹，里面有lib64、include还有一个不知道什么用的文档

tar -xvf cudnn-11.2-linux-x64-v8.1.1.33.tgz

1. 将解压文件放到指定位置
   将解压缩的文件放入对应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*

4.验证
查看cudnn版本信息：

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

如下图所示，可以看到已经安装成功cudnn8.1.1版本