conda公司的Tensorflow-gpu 2.5.0不承认GPU

Question

我制作了一个新的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

然后是蟒蛇

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

这产生了上述语句，它只识别我的CPU，而不是我的GPU。

Answer 1

预构建的二进制文件Tf2.5与CUDA 11.2和cuDNN 8.1兼容

请参阅经过测试的构建配置图https://www.tensorflow.org/install/source#gpu。

因此，您必须回回库达11.2，谢谢！