使用conda安装tensorflow-gpu 2.0

tensorflow1.x时, 使用conda安装tensorflow-gpu的最大好处是会自动安装cuda、cudnn等依赖。 现在2.0出来了, 但conda暂时还不支持其安装, 最新的版本仍是1.14。 若使用pip安装, 大概率会碰到报错:dynamic library 'libcudart.so.10.0' not found. 这是因为对应的cuda环境没装好。
可以手动安装对应的cuda版本, 也可以通过conda取巧:

conda install tensorflow-gpu=1.13
pip install tensorflow-gpu==2.0

首先通过conda安装1.13版本的tensorflow-gpu, 它会将2.0需要的cudart安装好。 然后再用pip安装tensorflow2.0覆盖conda环境中的tensorflow1.13。
需要注意的一点是, conda安装的tensorflow最好指定1.13版本, 它的cudart和2.0的相同。 而1.14的cudart则是10.1版本, 与2.0不兼容。 更早的版本则未测试过。

(2019.10.9, 亲测可用)

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值