ubuntu18安装GPU版本tensorflow

前言

记录在ubuntu18上安装GPU版本tensorflow经验。
主要参考:

  1. https://blog.csdn.net/tonydz0523/article/details/83109343
  2. 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可以同时安装多个版本。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值