来自 conda 的 Tensorflow-gpu 2.5.0 无法识别 GPU

Question

我新建了一个conda环境，从conda安装了tensorflow-gpu（最新版本是2.5.0）。然后，我测试了环境是否可以识别我的 GPU，但它没有。它returns

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 1364016363571256103
]

when 运行 list locality function on tensorflow。我错过了什么？

我在安装 tensorflow-gpu 时安装了 cuDNN 和 cudatoolkit 作为 conda 安装的依赖项。

cudnn==8.2.1.32
cudatoolkit==11.3.1

我运行的命令列表是：

conda create --name ML4
conda activate ML4
conda install tensorflow-gpu=2.5

然后在 python

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

上面的语句只识别我的 CPU 而不是我的 GPU。

Answer 1

TF 2.5 预构建二进制文件与 CUDA 11.2 和 cuDNN 8.1 兼容
请参阅测试构建配置图表 https://www.tensorflow.org/install/source#gpu。
因此你必须回滚到 cuda 11.2 谢谢！

Tensorflow-gpu 2.5.0 from conda doesn't recognise GPU