ubuntu 16.04 安装TensorFlow-GPU

@author: 向天笑
QQ:1500879657

1 安装前小贴士

贴士1:至今(2018.8.1)为止bantu 18.04对应打各种版本都试过,即使cuda安装成功,tensorflow也安装不成功,所以现阶段不建议安装18.04版本,而是安装16.04或14.04;

贴士2:目前(2018.8.1)tensorflow(tensorflow 1.9版本)最高支持到的CUDA版本是CUDA9.0,所以不要下载9.0以上版本。

贴士3:若你安装的是高版本的TensorFlow-GPU,不用降低自己的GCC到gcc-4.8版本,建议直接安装即可,若是gcc版本不对,则在安装nvidia driver环节会报错,那时再降低版本即可。

贴士4:在安装TensorFlow并允许时经常会出现:“ImportError: libcudnn.so.x(此处x表示版本号的一个整数): cannot open shared object file: No such file or directory”的错误。其原因是没有安装支持本机TensorFlow-GPU的cudnn或cudn所致,依据我的经验这种情况下属三个方法均可解决:

     方法一:换一个其它版本的TensorFlow-gpu(最方便,在pycharm上很容易实现)

     方法二:重新安装其它版本的cuda或cudnn(比较麻烦)

     方法三:设法安装一个和当前cuda或cudn版本可共存的其它cuda或cudn(至今未实现过2018.9.21)

贴士5:TensorFlow-GPU安装成功后,建议尽量不要更新系统显卡驱动(Ubuntu系统每个一定时间经常会弹出系统更新,此时建议忽略不要点击更新,因为一旦选择自动更新后显卡驱动也会随之更新,那么可能会更新到不适合自己系统的显卡驱动,很容易造成系统图像界面界面进不去或陷入重复输入账号密码的死循环中)同时关闭Ubuntu系统的更新功能(本人之前只要更新系统,大多数情况下会爆出各种或大或小的问题)。

 

2 安装显卡驱动

2.1查看自己打显卡:lspci |grep VGA

注意:在ubuntu上查看显卡适合的驱动理论上是可以的但是,我测试过上述命令,然并不容易看懂,在这里建议一下方法

2.1.1 方法一:首先查看自己电脑的牌子和型号,然后在京东、淘宝、中关村在线等平台上去查。

2.1.2 方法二:在Windows下查看自己电脑显卡驱动,再下载对应GPU驱动:可以看到更详细的信息。

2.2 在应nvidia观望查找适合电脑打显卡驱动并下载:驱动程序 | GeForce

 

3 安装cuda和cudnn

参考文章(本人安装时依第一个为主,第二个次之):

  (1)史上最全的ubuntu16.04安装nvidia驱动+cuda9.0+cuDnn7.0 - CSDN博客

(2)深度学习应用系列(一)| 在Ubuntu 18.04安装tensorflow 1.10 GPU版本

(3)真实机下 ubuntu 18.04 安装GPU +CUDA+cuDNN 以及其版本选择(亲测非常实用)

注意:安装cuda时加上命令--no-opengl-libs,不加这个选项会进入循环登陆。

例如:在终端运行指令 sudo sh cuda_8.0.27_linux.run --no-opengl-libs

 

4.安装与检测

参考文章:

CUDA安装和测试 - CSDN博客

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值