TensorFlow-gpu安装流程

TensorFlow-gpu安装流程

Windows安装TensorFlow-gpu版

建议
一、参考对应版本
https://blog.csdn.net/wanzhen4330/article/details/81660277
我的是win10,gpu版本需要安装cudnn6.0和cuda8.0
二、https://blog.csdn.net/yuejisuo1948/article/details/81043962、
三、重点 我推荐anconda 这个超级好用

Ubuntu16.04上升级pip版本

在装tensorflow之前,不管是不是最新的pip版本,都要更新一下,具体命令如下:
python 2.7版本:sudo pip install --upgrade pip
python 3.x版本:sudo pip3 install --upgrade pip
问题:ImportError: cannot import name main
修改/usr/bin/pip文件中
from pip import main
if name == ‘main’:
sys.exit(main())
改成
from pip import main
if name == ‘main’:
sys.exit(main._main())
可以解决这个bug。
配置完环境变量后 要source

TX2安装TensorFlow-gpu1.4.1

这个我用的是别人做好的轮子
链接:https://pan.baidu.com/s/1k5hZHIN6xqsX8U-g3TzZfA
提取码:nlru

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值