Tensorflow GPU не идентифицирует мои графические процессоры - PullRequest
Новая
       27

Tensorflow GPU не идентифицирует мои графические процессоры

0 голосов
/ 11 марта 2019

Я пытался использовать Tensorflow GPU, но, очевидно, Tersorflow не идентифицирует мои GPU.

Когда я бегу: 

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

В качестве вывода отображается только мой процессор. Я проверил все версии всего, и они кажутся совместимыми. У меня есть CUDA 10.1 с CUDA Toolkit, cuDNN 7.5 и Tensorflow 1.13.1. Я запускаю все на Ubuntu 18.xx

Что я делаю не так?

1 Ответ

0 голосов
/ 29 июня 2019

что такое выход: 

from tensorflow.python.client import device_lib

device_lib.list_local_devices()

В моей системе tensorflow не распознает GPU , потому что это XLA_GPU . Я не совсем уверен, почему XLA_GPU также не является GPU, кажется, что где-то в коде tenorflow-gpu отсутствует оператор OR.

Если в приведенном выше коде нет списка графических процессоров (а у вас есть): 

pip uninstall tensorflow
pip uninstall tensorflow-gpu
pip install tensorflow-gpu

... работал на меня.

...