Keras не использует GPU - как устранить неполадки? - PullRequest
Новая
       28

Keras не использует GPU - как устранить неполадки?

0 голосов
/ 15 мая 2018

Я пытаюсь обучить модель Keras на GPU, используя Tensorflow в качестве бэкэнда.

Я настроил все в соответствии с https://www.tensorflow.org/install/install_windows. Это мои настройки:

  • Я работаю в ноутбуке Jupyter в среде virtualenv.
  • В текущей среде virtualenv установлено tensorflow-gpu.
  • У меня установлены CUDA 9.1 и cudaDNN для CUDA 9.1.
  • cuDNN64_7.dll находится в местоположении, доступном через переменную PATH.
  • У меня на компьютере установлена ​​NVIDIA GeForce GTX 780 с последними драйверами.

Однако Tensorflow не видит ни одного пригодного для использования графического процессора: 

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

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 5275203639471190827
]

Керас также: 

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

[]

Как я могу отладить это?Как я могу узнать, где проблема?

1 Ответ

0 голосов
/ 16 мая 2018

Проверка
nvcc -V
и

nvidia-smi

и посмотрите, показывает ли он наш графический процессор или нет.

Предполагая, что вы cuda cudnn и все проверено, вам может понадобиться
1. Удалите keras
2. Удалите тензор потока
3. удалите tenorflow-gpu
4. Устанавливаем только tenorflow-gpu pip install tensorflow-gpu==1.5.0
5. Установите Keras сейчас.

Я выполнил эти шаги, и теперь Керас использует gpu.

Надеюсь, это поможет в некоторой степени.

...