获取cuDNN
去官网https://developer.nvidia.com/rdp/cudnn-download下载cuDNN,要注册NVIDIA账号。
我选择的是cuDNN v5.1 -> cuDNN v5.1 Library for Linux
安装cuDNN
下载cuDNN后解压
cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include #复制头文件
cd ../lib64
sudo rm -rf libcudnn.so libcudnn.so.5 #删除原动态文件
sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5 #生成软链接
sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接
若需要更换cudnn版本,则替换原来的libcudnn*,并重新软链接。
更新链接库:
sudo ldconfig
另外一个版本
从官网下载cudnn-8.0-linux-x64-v5.1.tgz for CUDA 8.0. 解压到当前目录:
tar -zxvf cudnn-8.0-linux-x64-v5.1.tgz
解压后的文件如下:
cuda/include/cudnn.h
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.5
cuda/lib64/libcudnn.so.5.1.5
cuda/lib64/libcudnn_static.a
然后执行:
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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*