來自 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 上運行列表位置 function 時。我錯過了什么？

我在安裝 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 謝謝！