首先,先去官网下载安装包。
https://developer.nvidia.com/rdp/cudnn-download
选择对于cuda驱动的版本。我下载的是cudnn v4版本。
下载完之后先解压缩:
tar -xvzf cudnn-7.0-linux-x64-v4.0-prod.tgz
然后:
$cd cuda/include
$sudo cp cudnn.h /usr/local/include/
$cd ../lib64
$sudo cp lib* /usr/local/lib/
$cd /usr/local/lib
$sudo chmod +r libcudnn.so.4.0.7
$sudo ln -sf libcudnn.so.4.0.7 libcudnn.so.4
$sudo ln -sf libcudnn.so.4 libcudnn.so
$sudo ldconfig
安装完毕