Ubuntu16.04 cuda8.0+cudnn6.0+python3.5 配置tensorflow-gpu 1.4.0

机子情况

win10+ubuntu16.04双系统
cpu:intel i7-7700
gpu:GTXI060 6G EVGA
在ubuntu16.04上想配置GPU版的tensorflow
备注:
我原来系统是win10,后来用u盘制作启动盘安装了ubuntu16.04 LTS

安装驱动

先去官网下自己显卡型号对应的驱动,不赘述

卸载掉原有驱动

sudo apt-get purge nvidia*

禁用nouveau

我不太清楚双系统是不是有什么特殊性,我试了几个常见的方法都不行。最后是以下这个方法奏效了
参考链接:https://blog.csdn.net/u014797226/article/details/79626693
1、打开编辑配置文件:

sudo gedit /etc/modprobe.d/blacklist.conf

2、在最后一行添加:

blacklist nouveau

3、改好后执行命令:

sudo update-initramfs -u

4、重启使之生效:

sudo reboot

5、检查是否成功

lsmod | grep nouveau

没有任何输出即成功

开始安装驱动

我按照上面那个参考链接的建议做法试了,会报错,说什么x服务仍在运行,还有一个错误不懂说什么。
然后我自己是这样试成功的:
给予安装文件权限

sudo chmod a+x NVIDIA-Linux-x86_64-390.59.run     (下载的驱动文件名)

然后

sudo ./NVIDIA-Linux-x86_64-390.25.run --no-opengl-files --no-x-check --no-nouveau-check

-–no-opengl-files 只安装驱动文件,不安装OpenGL文件。这个参数最重要
--no-x-check 安装驱动时不检查X服务
--no-nouveau-check 安装驱动时不检查nouveau
参考链接里头说,后面两个参数可不加。但我试过后发现,全部加上才行

注意no那里是双杠
重启,没有问题,输入命令:nvidia-smi
显示出显卡对应信息,则安装成功

1.添加conf 文件:cat /etc/modprobe.d/blacklist-nouveau.confblacklist nouveauoptions nouveau modeset=0
2. 重新生成 kernel initramfs:sudo update-initramfs -u
3. 3. 重启:sudo reboot

参考链接:
https://www.zhihu.com/question/68565667/answer/264871312

http://www.cnblogs.com/Qwells/p/6086773.html#undefined

https://blog.csdn.net/zafir_410/article/details/73188228?utm_source=itdadao&utm_medium=referral

安装CUDA8.0

注意一点:

在安装cuda自带的驱动时,一定要选N,否则之前的就白装了

安装CUDNN6.0

查看是否安装好

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

安装Anaconda

https://blog.csdn.net/u012318074/article/details/77074665
官网下载
使用cd命令切换到下载目录
使用ls命令列出文件目录
bash Ana
按tab键,回车

进入tensorflow环境

激活

conda activate tensorflow-1.4.0

退出

conda deactivate

测试安装是否成功

在Pycharm设置tensorflow的环境变量

terminal里面是可以用的,但是不太方便,安装了pycharm后,发现每次新建一个使用tensorflow的py文件都要在debug设置那里加环境变量:

export LD_LIBRARY_PATH=/usr/local/cuda/lib64/
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64/
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值