cuDNN是针对深度神经网络专门开发的cuda的计算库,其安装倒不是什么大问题,下载倒是挺折腾的,又是注册又是填表格的,还是外网,往往填着填着就没反应了,如此反复折腾几次,只好上知乎找大佬们的链接,
链接为:cudnn-8.0-linux-x64-v5.1.tgz
密码:kzcf
安装很简单,就是先解压再安装:
sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
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版本
sudo cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A2
具体图解过程可以参考以下链接:
点击打开链接 (讲的很清晰简明,首选方案)
点击打开链接 (这个解释的比较清楚,对常见错误也有介绍,可参考翻阅)