tensorflow 安装

2019-3-26 10:45:56
ubunutu上切换 cuda版本很方便,
参考
https://blog.csdn.net/haithink/article/details/88713218
并且切换了 cuda后, tensorflow python版本依然能正常使用

2019-3-7 10:57:40
更新,按照官方的说法, tensorflow 1.5 及以后都需要cuda 9.0
但 有些时候环境中的cuda不是 自己能决定的,
需要在 cuda 8.0 上面安装 tensorflow1.8.0 或者 1.10.0 这些比较高的版本

之前 没有记录下来

方法如下:

创建一个环境
conda create -n tensorflow python=3.5

直接
conda install tensorflow-gpu==1.8

这里不要使用 anaconda的 channle,
随后conda会列出将要安装的包,如果没有 cuda 9.0之类的包 就基本要成功了

https://anaconda.org/conda-forge/tensorflow

conda install -c conda-forge tensorflow

好像还是有些坑, 慢慢记录下,每次装完 都不注意总结,明明安装过程中会出现一些坑。。。。

进入python环境,使用如下代码 检测 是否tensorflow 可以检测到 GPU

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

如果检测不到 GPU,那么 使用 PIP 安装,使用豆瓣源, 也会比较快

pip install tensorflow-gpu==1.9.0 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值