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