Tensorflow-gpu Ubuntu16.04 GTX1060 双显卡

Tensorflow1.4.0不支持cuda9.0

一开始在cuda官网上下载了cuda9.0,非常不幸,一番辛苦下来发现现在最新版本的tensorflow-gpu不支持cuda9.0。因此,只能在网上寻找cuda8.0进行下载

卸载已有的cuda

尝试了网上很多已有的命令,这个命令可以使用

sudo apt-get autoremove --purge cuda

如果没有卸载干净,可以使用

rm -rf <文件夹路径>

将剩余的整个文件夹删除(包括文件夹下的所有文件)

安装Anaconda2

anaconda2是一个非常好的集成环境,里面包含了python,以及spyder IDE,PIP等工具。注意安装好anaconda2后需要配置环境变量。

pip 安装 tensorflow

使用

pip install tensorflow-gpu

安装 tensorflow的gpu版本

下载cuda8.0

推荐寻找百度网盘

下载cudnn6 for cuda 8.0

注意,截至2017-12-1, 当前版本的tensorflow1.4.0只是支持cudnn6,不要下成cudnn7

安装cuda8.0和cudnn6

参考博文
http://blog.csdn.net/zhaoyu106/article/details/52793183
亲测没有问题,但是使用cudnn6的话,gcc不需要降低版本。

安装cuda8.0,使用.run文件
sudo sh cuda_8.0.44_linux.run

我是在已经安装了nvidia驱动的情况下进行安装的,因此,提示是否安装nvidia driver的时候,我选择了no。其他都是一路accept,yes,或者回车

安装cudnn6.0

cudnn其实类似于cuda的扩展,下载下来的是一个压缩文件,解压之后进行下面三行命令即可。(解压出的文件默认名是cuda)

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*

Nvidia 显卡驱动安装

可以参考
http://blog.csdn.net/qq_33843297/article/details/78683569

另外有个命令暂时这里保存一下

sudo service lightdm stop  #用于关闭图形化界面
sudo service lightdm restart  #用于开启图形化界面

使用ctrl+Alt+F1可以从linux的图形界面进入命令行界面
使用ctrl+Alt+F6可以从linux的命令行界面返回图形界面

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值