官网cudnn库链接:
https://developer.nvidia.com/rdp/cudnn-archive
请选择符合系统、CUDA配置的版本。
推荐使用solitairetheme8类型下载,步骤如下:
①修改安装包后缀名为.tgz
②tar -zxvf 命令解压
tar -zxvf cudnn-<CUDA version>-linux-x64-<version>.tgz;
③将头文件cudnn.h复制到cuda对应的include目录下
cp cudnn.h ../cuda/include;
④将lib库(libcudnn*)复制到cuda对应的lib64目录下
cp cudnn.h ../cuda/include;
*(3、4两步可能需要sudo权限,且建议先cd到所在目录再执行复制)
⑤赋予所有用户读取头文件和lib库的权限
chmod a+r ../cuda/include/cudnn.h;
chmod a+r ../cuda/lib64/libcudnn*
⑥查看cudnn版本进行验证
cat ../cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2;