一、在NVIDIA官网下载cuDNN
1、网址:cuDNN Archive | NVIDIA 开发者
2、由于本人CUDA安装的是11.8所以下载Download cuDNN v8.9.3 (July 11th, 2023), for CUDA 11.x
进行安装
3、我们选择Local Installer for Linux x86_64 (Tar)下载
4、解压tar -xvf cudnn-linux-x86_64-8.9.3.28_cuda11-archive.tar
二、安装
1、将cudnn的头文件和库文件拷贝到CUDA目录下,方便程序调用
sudo cp cudnn-linux-x86_64-8.9.3.28_cuda11-archive/include/cudnn*.h /usr/local/cuda-11.8/include
sudo cp cudnn-linux-x86_64-8.9.3.28_cuda11-archive/lib/libcudnn* /usr/local/cuda-11.8/lib64
sudo chmod a+x /usr/local/cuda-11.8/include/cudnn*.h
sudo chmod a+x /usr/local/cuda-11.8/lib64/libcudnn*