在安装完cuda之后,根据版本的提示,安装对应的cudnn即可,下载网址https://link.youkuaiyun.com/?target=https%3A%2F%2Fdeveloper.nvidia.com%2Frdp%2Fcudnn-archive
下载之后是一个压缩包,解压并执行一下指令:
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 /usr/local/cuda/lib64/libcudnn*
最后验证我们是否正确安装cudnn, 执行以下指令:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果成功的话会出现以下界面
如果不出现任何指令的话,大概率是因为是安装的cudnn版本过高,将下载的cudnn卸载再重新下载一个版本低一点的就好了。
卸载cudnn代码:
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn