深度学习环境搭建(四)之 CUDNN安装

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

1.下载CUDNN

访问官网链接,根据CUDA版本和Ubuntu版本选择CUDNN的版本。前面使用的是CUDA 10.1,系统的版本是Ubuntu 20.04,CUDA 11.04,因此这里选择相对应的CUDNN v8.2.4.
深度学习环境搭建(四)之 CUDNN安装_第1张图片
注意到这里有很多不同格式的CUDA,相应的安装方法也不同。下面我们分别介绍。

2. CUDNN安装(tgz格式)

选择cuDNN Library for Linux (x86)进行下载
深度学习环境搭建(四)之 CUDNN安装_第2张图片
打开终端,解压文件:

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

如果出现下图的信息,则说明安装成功:
深度学习环境搭建(四)之 CUDNN安装_第3张图片

你可能感兴趣的:(深度学习,深度学习)