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

1.下载CUDNN

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

CUDNN

注意到这里有很多不同格式的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

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

  
  
  安装成功