简体   繁体   中英

tensorflow cannot find GPU

I had install "tensorflow-GPU", CUDA 10.0. and my GPU is GTX1660 ti. I also tested bu CUDA 10.2 and 11. I added cudnn to windows PATH but I still got this error.

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

I found the problem. the problem was versions of CUDA and cudnn.

在此处输入图像描述

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM