张量流找不到GPU

时间:2020-07-02 16:18:24

标签: python tensorflow gpu

我安装了“ tensorflow-GPU”,CUDA 10.0。我的GPU是GTX1660 ti。 我还测试了CUDA 10.2和11。我在Windows PATH中添加了cudnn,但仍然出现此错误。

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

1 个答案:

答案 0 :(得分:0)

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

enter image description here