ubontu--cuDNN安装

时间:2024-11-14 09:47:12

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_MAJORCUDNN_MINORCUDNN_PATCHLEVEL 定义的具体数值,就说明系统可以正常识别 cuDNN 库文件,配置已经正确完成。

5.