Ubuntu18安装GPU版tensorflow
前言
记录在ubuntu18上安装GPU版本tensorflow经验。
主要参考:
- https://blog.csdn.net/tonydz0523/article/details/83109343
- https://blog.csdn.net/u014670893/article/details/82467384#CUDA%20Toolkit%C2%A09.0
要点与碰到的坑
1.显卡驱动版本不能太高,推荐的版本都是最新的,并不合适,需要参考网上别人的经验贴。
2.CUDA不要安装10.1!到目前(2019.11.17)为止,10.1版本与tensorflow1.×版本是不匹配的,所以安装10.0版本。
安装步骤
应为参考博客里已经讲的很清楚了,不再赘述,只添加我碰到的坑与解决办法。
1安装显卡驱动、cuda、cudnn
参考这篇博客安装显卡驱动,注意不要安装推荐的驱动版本,尽量低些,我的显卡是RTX2080Ti,第一次安装430版本不兼容,第二次安装了418版本OK。如果不用命令行方式也可以在NVIDIA官网下载安装文件。下载好.run文件后用sudo chmod 777 ./NVIDIA**.run添加执行权限,然后在ctrl+f5进入命令行模式,运行这个run文件进行安装。这篇博客里提到的gcc或者g++降级不用管,因为10.0版本可以使用ubuntu18的7.3版本gcc/g++。
参考这篇博客安装cuda与cudnn,注意cuda要选择10.0版本(deb安装包即可)。同时安装cudnn,下载安装7.6.0即可。
2安装tensorflow-gpu
tensorflow版本跟cuda是有对应关系的,对于cuda10.0,安装tensorflow1.13.1即可,太低不行,tensorflow2.0也不行。
可能遇到的bug
无法打开libcublas.so.10文件,没有这个文件或目录:我碰到这个问题是因为一开始安装了cuda10.1,换成cuda10.0就好了。而且不需要卸载cuda10.1,因为cuda可以同时安装多个版本。