Tensorflow 2.3.0 CUDA Toolkit версии 10.1 не использует графический процессор - PullRequest
0 голосов
/ 04 августа 2020

У меня была работа с tenorflow 2.0 с моим графическим процессором RTX2070. Я сделал обновление windows, чтобы использовать tf-nightly. Не понравилось, поэтому удалил его и переустановил tensorflow 2.3.0. Выполнил предыдущий код python, который раньше работал нормально с GPU, но не использовал GPU. Много чего перепробовал. Наконец-то все началось. Переустановил Anaconda, создал новую среду. Удалил Cuda toolkit 10.1 и переустановил его. Установлен cuDnn SDK 7.6 в каталог c: \ Tools. Проверенная переменная env пути для включения

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include;
C:\tools\cuda\bin;%PATH%
       #then ran this code:
import tensorflow as tf
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
print(tf.__version__)
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
tf.test.is_gpu_available()
     #I get the result
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 15177607927005893519
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 4640072765546557805
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 16675502319763286567
physical_device_desc: "device: XLA_GPU device"
]
2.3.0
Num GPUs Available:  0

False

tensorflow still does not use GPU. What an I missing? 

also same problem using python 3.7.0 and same problem using tensorflow 2.0.0


1 Ответ

0 голосов
/ 05 августа 2020

Я обнаружил, что могу заставить тензорный поток распознавать графический процессор, если в моей рабочей среде с использованием conda я запускаю conda install cudnn == 7.6.4, который работает с CUDA 10.1.0, результирующие сообщения в приглашении anaconda:

...