官网下载需要的cudnn版本:https://developer.nvidia.com/rdp/cudnn-archive 下载需要注册账号 我安装的是cuda8.0,对应cudnn6.0
下载后解压,进入 cudnn目录,有include和lib64两个文件夹
复制include中的cudnn.h文件:
cp cuda/include/cudnn.h/usr/local/cuda-8.0/include/
在/usr/local/cuda-8.0/lib64 中删除:
rm libcudnn.so.6 libcudnn.so.6.0.21
复制lib64中的:
cp libcudnnn.so.6.0.21 /usr/local/cuda-8.0/lib64/
重新建立软连接,在/usr/local/cuda-8.0/lib64中:
ln -s libcudnnn.so.6.0.21 libcudnn.so.6
针对错误:ImportError: /usr/local/cuda/lib64/libcudnn.so.6: file too short
如果软连接建立错误,会出现以上报错
错误:ImportError: libcusolver.so.8.0: cannot open shared object file: No such file or directory
export LD_LIBRARY_PATH=/usr/local/cuda/lib64/