tensorflow-gpu 2.3 安裝了 cuda 並且 cudnn 無法檢測到 GPU？

Question

我有一個關於在 tensorflow 中使用 GPU 的問題我正在使用 RTX2060

我正在關注本教程解決方案 TensorFlow 2 not using GPU

我已經安裝

cudatoolkit = 10.2.89

cudnn = 7.6.5

張量流 GPU = 2.3.0

但是當我簽入 python 時

print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

原來我的 GPU num 是 0

我已經嘗試在整個互聯網上找到解決方案，但仍然無法修復它。

請幫忙，謝謝。

Answer 1

因此，該指南不足以使用 GPU 設置 tensorflow； 例如，這里有一個隨機指南（其中有很多），解釋了一些缺少的組件。 請注意處理 CUDA 驅動程序、適當的 GPU 驅動程序版本、特定 linux 發行版（Ubuntu 20.04）等的部分：

如何設置 tensorlfow 以與 GPU 一起使用的隨機示例

我不知道這是否是一個好的指南，或者它是否有效，但我只是想指出您正在使用的指南中一些重要的缺失部分。

So you have to set up the CUDA driver (I noticed that you don't mention a CUDA driver version at all) with the appropriate GPU drivers and the appropriate linux distribution (I have never set up CUDA on Windows so idk about that) in訂購 tensorflow 以便能夠識別您的 GPU。