1. 下载 cuDNN
https://developer.nvidia.com/cudnn
2. 拷贝到服务器/home/<username>文件夹下
解压缩到当前文件夹:
tar -xvf cudnn-linux-x86_64-9.5.1.17_cuda11-archive.tar.xz
复制头文件和库文件到cuda安装目录/usr/local/cuda/
sudo cp /home/username/cudnn-linux-x86_64-9.5.1.17_cuda11-archive/include/cudnn*.h /usr/local/cuda/include/
sudo cp /home/username/cudnn-linux-x86_64-9.5.1.17_cuda11-archive/lib/libcudnn* /usr/local/cuda/lib64/
3.设置文件权限
sudo chmod a+r /usr/local/cuda/include/cudnn*.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
4. 检验是否成功安装
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
能看到 CUDNN_MAJOR
、CUDNN_MINOR
和 CUDNN_PATCHLEVEL
定义的具体数值,就说明系统可以正常识别 cuDNN 库文件,配置已经正确完成。
5.