CUDNN下载地址
选择箭头所指版本进行下载并解压。
解压完成后,打开终端
cd cudnn-10.2-linux-x64-v8.1.1.33/cuda/
sudo cp include/cudnn.h /usr/local/cuda-10.2/include/
sudo cp lib64/libcudnn* /usr/local/cuda-10.2/lib64/
sudo cp include/cudnn_version.h /usr/local/cuda-10.2/include/
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*
注意自己cuda的版本
验证
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
修改cudnn软连接
cd /usr/local/cuda-10.2/lib64/
sudo rm libcudnn.so.8
sudo ln -s libcudnn.so8.1.1 libcudnn.so.8
上边命令要对应自己的版本号进行修改。例如,如果你是cudnn版本是7.6.5,那么命令就是
sudo rm libcudnn.so.7
sudo ln -s libcudnn.so.7.6.5 libcudnn.so.7