windows 10 Tensorflow GPU 版本安装

19 篇文章 0 订阅
11 篇文章 0 订阅

时隔一年,再装tensorflow,发现windows10上面安装已经比之前简单了很多

https://www.tensorflow.org/install/gpu?hl=zh-cn
现在GPU 版本和 CPU版本已经合并了,所以可以直接安装
pip install tensorflow

碰到错误,可能是因为之前装 tensorflow-gpu失败导致的
https://www.cnblogs.com/conver/p/11141176.html

Cannot uninstall ‘wrapt’. It is a distutils installed project
https://www.cnblogs.com/conver/p/11141176.html

直接:
pip install -U --ignore-installed wrapt enum34 simplejson netaddr

然后装CUDA 和 cudnn
CUDA: cuda_10.1.105_418.96_win10.exe
CUDNN: cudnn-10.1-windows10-x64-v7.6.5.32
前者下载exe,一直点下一步,如果要自己选安装目录,选择自定义安装
然后把 bin目录路径添加到系统PATH环境变量中,貌似需要重启实现环境变量生效

后者的安装则是拷贝cudnn中的文件到对应的cuda安装目录中即可,主要是版本问题,
对应cuda10.1 , cudnn版本不对不会影响 python中 import tensorflow,但运行时会报错,试了一些早期版本发现不行, 最后用了 cudnn-10.1-windows10-x64-v7.6.5.32才成功,

检验;

import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())

tensorflow版本为2.3.1

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值