tensorflow 找不到 GPU

Question

我已经安装了“tensorflow-GPU”，CUDA 10.0。 而我的 GPU 是 GTX1660 ti。 我还测试了 bu CUDA 10.2 和 11。我将 cudnn 添加到 windows PATH 但我仍然收到此错误。

Traceback (most recent call last):
  File "D:\Electrical And Programming office\6_AI\RL\5_Pong_DQN\pong_DGN.py", line 38, in <module>
    gpu = tf.config.experimental.list_physical_devices('GPU')[0]
IndexError: list index out of range

>>> print(tf.test.is_built_with_cuda())
True

Answer 1

我发现了问题。 问题是 CUDA 和 cudnn 的版本。