cuDNN 安装
2017-03-01 本文已影响350人
盲狙小堡垒
前提
已经安装好了CUDA,并且知道cuda目录(一般是/usr/local/cuda)
如果不知道可以用以下命令查询
which nvcc
准备文件
下载你需要的版本:https://developer.nvidia.com/rdp/cudnn-download
例如我下载cnDNN5(基于CUDA 8.0 和 Linux)(此操作不一定要在cuda安装目录下,URL地址需要根据你的版本来替换)
wget https://developer.nvidia.com/rdp/assets/cudnn-7.5-linux-x64-v5.0-ga-tgz
tar -xvf cudnn-7.5-linux-x64-v5.0-ga-tgz
安装(复制文件+创建软连接)
其实cuDNN的安装就是把几个文件复制到which nvcc显示的目录下。把解压好后的cuda里面的include 和 lib64 都复制到你的CUDA目录下(一般是/usr/local/cuda)
然后重建一下软连接。
ln -s libcudnn.so.5.1.3 libcudnn.so.5.1
ln -s libcudnn.so.5.1 libcudnn.so
OK, now We are done!