win10安装tensorflow-gpu和keras

本文配置

python 环境:python 3.6
cuda 版本号:cuda_10.0.130_411.31_win10
cudnn 版本号:cudnn-10.0-windows10-x64-v7.6.5.32
Tensorflow 版本号: tensorflow-gpu 1.14.0
Keras 版本号:keras 2.2.5

写在前面

在电脑里找到NIVIDIA Control Pannel查看你的电脑能支持的cuda最高版本,如果安装了高于此版本的cuda,即使安装成功也无法正常使用。

比如,如果按照tensorflw-cpu版本的方法来安装gpu版本,即,在虚拟环境中:conda install tensorflow-gpu 在我这里显示会直接安装tensorflow2.1 + cuda10.1.243+cudnn7.6.5,就会导致明明安装成功但tensorflow使用时出错的问题。

另外,确定了cuda版本之后,后续安装的 cuda、cudnn、Tensorflow、keras 的版本号务必匹配,否则会发生意想不到的错误!!!

TensorFlow和cuda以及cudnn版本匹配查询
https://tensorflow.google.cn/install/source_windows
Tensorflow 与Keras版本匹配查询
https://docs.floydhub.com/guides/environments/

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值