cuDNN 8.0.4 for CUDA 11.1.0
1.下载安装包https://developer.nvidia.com/rdp/cudnn-archive
(cuDNN Library for Linux(x86_64))
得到cudnn-11.1-linux-x64-v8.0.4.30.tgz
2.到压缩包所在目录打开终端,解压缩
tar -zxvf cudnn-11.1-linux-x64-v8.0.4.30.tgz
得到cuda文件夹,里面包括include和lib64两个文件夹
3.将cuda中的include中的文件复制到/usr/local/cuda-11.1/include
将cuda中的lib64中的文件复制到/usr/local/cuda-11.1/lib64
sudo cp cuda/include/* /usr/local/cuda-11.1/include
sudo cp cuda/lib64/* /usr/local/cuda-11.1/lib64
4.添加权限
sudo chmod a+r /usr/local/cuda-11.1/include/cudnn.h /usr/local/cuda-11.1/lib64/libcudnn*
5.查看cuDNN信息
cat /usr/local/cuda-11.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
这里的cuda文件夹就是下载下的cuDNN.