Tensorflow无法检测到Nvidia GPU。

Question

你好，我正在尝试让TensorFlow V2.11找到我的eGPU（RTX 3060 Ti），但是遇到了困难。

我当前使用的系统是Windows 11，CUDA版本是12。根据建议，我正在下载CUDA 11和CUDnn。

我已经尝试过以下代码：

import tensorflow as tf
tf.config.list_physical_devices('GPU')

输出结果为：

[]

非常感谢您的帮助。

- Peterdagrape

---

tf.test.is_built_with_cuda() 返回什么？ - talonmies

变得更有趣了，我听说2.11与Windows上的GPU不兼容，所以降级到2.10，并运行该命令，我得到True，但是导入库告诉我它找不到cuda dll，目前正在努力弄清楚它在哪里寻找。 - Peterdagrape

Answer 1

Tensorflow 2.11不支持Windows机器上的GPU。 TensorFlow 2.10是最后一个在本机Windows上支持GPU的TensorFlow版本。因此，您可以尝试安装Tensorflow 2.10进行GPU设置。

此外，您需要在系统中安装特定版本的CUDA和cuDNN以支持GPU，这是为Tensorflow 2.10（Tensorflow> = 2.5）提供的CUDA 11.2和cuDNN 8.1。

请按照链接中提到的硬件/软件要求检查并在安装这些软件后将路径设置为bin目录。

现在，请按照同一链接中提到的逐步说明进行操作，并使用下面的代码验证GPU设置。

python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

- user11530462

---

是的，我发现了这个问题。我曾经苦苦寻找2.10版本的CUDA dlls，但我认为我现在做了一些让它正常工作的事情。 - Peterdagrape