以Tar包的方式安装cuDNN 8

本文记录了cuDNN 8的安装过程。

环境信息

• Ubuntu Server 20.04.4 LTS
• CUDA 11.3.1

准备工作

在正式安装开始前，请先确认以下内容。

安装CUDA

已完成对应版本的CUDA Toolkit的安装。

安装zlib

sudo apt-get install zlib1g

下载安装文件

从官网获取安装文件。

cuDNN Archive

NVIDIA cuDNN is a GPU-accelerated library of primitives for deep neural networks.

NVIDIA Developer

安装

下载完成后，正式开始安装。

进入到安装文件下载目录，将其解压缩。

tar -xvf cudnn-11.3-linux-x64-v8.2.1.32.tgz

然后拷贝头文件到CUDA目录。

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include

再拷贝库文件到CUDA目录。

sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64

最后赋予文件读取权限。

sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

至此安装完成。

接下来即可安装PyTorch与TensorFlow等深度学习框架。