cuda11.5对应版本的TensorFlow-gpu到底是多少？

Question

各位好！本人在配深度学习的环境，已经安装好了pytorch（GPU版本）和cuda，现在在安装TensorFlow-GPU遇到了难题，我的cuda是11.5的，找不到对应版本的TensorFlow-gpu。

Answer 1

阿里嘎多学长整理AIGC生成，因移动端显示问题导致当前答案未能完全显示，请使用PC端查看更加详细的解答过程

问题解决方案

根据TensorFlow的官方文档，TensorFlow支持的CUDA版本如下：

• TensorFlow 2.x：支持CUDA 11.0、11.1、11.2、11.3、11.4、11.5
• TensorFlow 1.x：支持CUDA 9.0、9.1、9.2、10.0、10.1、10.2、11.0、11.1、11.2

因此，根据你的描述，你的CUDA 11.5版本是支持TensorFlow 2.x的。

具体来说，你可以尝试安装TensorFlow 2.x的GPU版本，例如：

pip install tensorflow-gpu==2.10.0

或者，你也可以尝试安装最新的TensorFlow 2.x版本：

pip install tensorflow-gpu

如果你遇到任何问题，请尝试检查你的CUDA和cuDNN版本是否正确安装，并且检查你的TensorFlow安装是否成功。