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

物联沃分享整理
物联沃-IOTWORD物联网 » TensorFlow、TensorFlow-GPU、CUDA、cuDNN、Python和NumPy对应版本指南

发表评论