一、下载cudnn
下载下图所示的文件
如果是需要和cuda10版本所对应的cudnn的话,可以直接百度网盘自取
链接:https://pan.baidu.com/s/1iibyinbONNNRUXjLmlWccg
提取码:zclp
复制这段内容后打开百度网盘手机App,操作更方便哦
二、安装cudnn
首先解压缩下的cudnn压缩包文件
tar -xzvf 刚刚下载的压缩包名.tgz
拷贝头文件和库文件并给予权限
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
三、验证安装是否成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果显示如下,即安装成功
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)