安装完CUDA Driver和CUDA后，还需要安装CUDNN，NVIDIA用于深度学习GPU加速的库。

1.下载CUDNN

访问官网链接，根据CUDA版本和Ubuntu版本选择CUDNN的版本。前面使用的是CUDA 10.1，系统的版本是Ubuntu 20.04，CUDA 11.04，因此这里选择相对应的CUDNN v8.2.4.

注意到这里有很多不同格式的CUDA，相应的安装方法也不同。下面我们分别介绍。

2. CUDNN安装（tgz格式）

选择cuDNN Library for Linux (x86)进行下载

打开终端，解压文件:

tar -xzvf cudnn-10.1-linux-x64-v7.6.5.32.tgz

将解压后的文件拷贝至指定目录并赋予权限:

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 /usr/local/cuda/lib64/libcudnn*

3. CUDNN安装（deb格式）

选择如下三个文件进行下载

  cuDNN Runtime Library for Ubuntu 16.04 (Deb)
  cuDNN Developer Library for Ubuntu 16.04 (Deb)
  cuDNN Code Samples and User Guide for Ubuntu 16.04 (Deb)

打开终端找到下载的deb 文件所在的位置，然后按步骤安装三个文件

sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64.deb

4. 检验是否安装成功

打开终端输入

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

如果出现下图的信息，则说明安装成功：