1. 查看 cuda 和 cudnn 版本
命令:
cat /usr/local/cuda/version.txt
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
2. 安装
从官网(https://developer.nvidia.com/rdp/cudnn-download)上下载需要的安装包:cudnn-8.0-linux-x64-v6.0.tgz
解压处理:tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz
解压后得到一个cuda文件夹:
进入cuda文件夹,里面有:一个include文件夹,和一个lib64文件夹,复制一些东西到我们之前安装cuda的路径下(/usr/local/cuda/)
命令:
$cd cuda
$sudo cp lib64/lib* /usr/local/cuda/lib64/
$sudo cp include/cudnn.h /usr/local/cuda/include/
3. 更新cuDNN库文件的软连接
命令:
$ sudo chmod +r libcudnn.so.6.0.21
$ sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6