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

【问题标题】：Tensorflow-gpu 2.5.0 from conda doesn't recognise GPU来自 conda 的 Tensorflow-gpu 2.5.0 无法识别 GPU
【发布时间】：2021-10-24 23:46:48
【问题描述】：

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

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

在 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。

【问题讨论】：

【解决方案1】：

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

【讨论】：