Ubuntu 更换 cudnn 版本

由于 cuda tensorflow-gpu cudnn 等版本号对应问题，在运行tensorflow-gpu时会出现错误，此时应到官网下载对应的cudnn版本，本文简要介绍更换cudnn版本号的方法：

第一步：官网下载cudnn（如cuda 9.0 对应 cudnn 7.6.2），解压出cuda文件夹

第二步：删除已有cudnn:

$ sudo rm -rf /usr/local/cuda/include/cudnn.h

$ sudo rm -rf /usr/local/cuda/lib64/libcudnn*

第三步：cd到解压的cuda文件夹下复制对应的文件到路径：

$ sudo cp include/cudnn.h /usr/local/cuda/include/

$ sudo cp lib64/lib* /usr/local/cuda/lib64/

第四步：添加文件路径，注意根据自己的版本号添加，本文以cudnn 7.6.2为例

$ sudo chmod +r libcudnn.so.7.6.2

$ sudo ln -sf libcudnn.so.7.6.2 libcudnn.so.7

$ sudo ln -sf libcudnn.so.7 libcudnn.so

$ sudo ldconfig

更换完成！再次运行代码，报错消失

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！