tensorflow,tensorflow-gpu, CUDA, cuDNN,Python, numpy对应版本

本机GPU是NVIDIA GeForce GTX 1650, 支持的最高版本CUDA是12.0。想要在Windows下使用GPU跑深度学习模型,需要使用tensorflow-gpu,其目前最高版本是2.6.0。所以其他依赖环境均按照tensorflow-gpu的需求来配置。

对应版本如下

Tensorflow-gpu 2.6.0 (目前为止最高版本)
Python 3.9.18 (能够支持tensorflow-GPU的最高Python版本)
Tensorflow 2.6.0
CUDA 11.2.2 (装最高版本12.0就没有对应的tensorflow-gpu版本了…)
cuDNN 8.2.1 (8.1 for CUDA 11.x的不行,已经踩过坑了。)
numpy 1.23.4 (numpy版本不对的话,会导致import tensorflow as tf 这一步报错,错误信息是 …numpy has no attribute object…)
(踩坑的血泪教训…)
##参考链接
https://developer.baidu.com/article/detail.html?id=1804196

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值