一、安装 cuda

1. 首先查看自身电脑最高支持的cuda版本为多少，在终端输入以下指令

nvidia-smi

可以看到我的最高支持 cuda 11.4 。

2. 打开英伟达官网 下载官方cuda

此处我下载的是cuda 11.0.3 版本，并选择相应配置，复制指令至终端下载

3. 运行官网安装指令

 4. 依次选择continue和accept 点击确认

 5. 按回车取消Driver项的安装，然后回车选择Install

 6. 添加环境变量
     gedit ~/.bashrc打开文件，在文件结尾添加如下语句：

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

7.  保存、更新环境变量

source ~/.bashrc

8. 输入nvcc -V查看版本号

二、 安装 cudnn

1. 进入官网 ，寻找合适的 cudnn 版本，下载对应版本的cudnn 的 runtime developer 和samples

2. 下载得到四个文件

3. 安装上述软件包

tar -zxvf cudnn-11.0-linux-x64-v8.0.4.30.tgz
sudo cp -rf cuda/include/cudnn* /usr/local/cuda/include/
sudo cp -rf cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
 
sudo dpkg -i libcudnn8_8.0.4.30-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.4.30-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.4.30-1+cuda11.0_amd64.deb

 4. 测试

拷贝例程到用户目录

cp -r /usr/src/cudnn_samples_v8/ $HOME

5. 转到minist例程

cd  $HOME/cudnn_samples_v8/mnistCUDNN

6. 编译例程

sudo make clean && make

7. 运行例程

./mnistCUDNN

8. 如果cudnn被恰当的安装，将输出打印消息：Test passed!

9. cudnn 检测版本号

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