Keras не использует мой графический процессор Nvidia при обучении нейронной сети - PullRequest
Новая
       18

Keras не использует мой графический процессор Nvidia при обучении нейронной сети

0 голосов
/ 07 мая 2019

Мой графический процессор не используется Keras / TensorFlow.

Чтобы попытаться заставить мой GPU работать с tenorflow, я попытался установить tenorflow-gpu (я использую Python 3.6.8 в Windows): 

pip3 install tensorflow-gpu --user

python -m notebook

import tensorflow as tf

Я получил тогда следующие ошибки: 

ImportError ... Traceback (most recent call last),

ImportError: DLL load failed: Le module spécifié est introuvable.

ImportError ... Traceback (most recent call last)

Затем я делаю pip3 install tensorflow, python - notebook, а затем import tensorflow as tf работает, но когда я продолжаю: 

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

print('Tensorflow: ', tf._ _ version _ _)

И я получаю: 

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

Это означает, что графический процессор не найден (и у меня на Lenovo Thinkpad P72 есть Intel UHD Graphics P630 и Nvidia Quadro P5200).

Также обратите внимание, что когда я запускаю! Nvidia-smi, я вижу, что Nvidia обнаруживается (как GPU [0]), поэтому я не понимаю, почему Tensorflow / Keras не используют его при обучении нейронных сетей.

Спасибо за помощь.

1 Ответ

0 голосов
/ 07 мая 2019

Я думаю, что решил проблему:)

После установки Anaconda3 я продолжил в PowerShell Anaconda следующим образом:

conda create -n tenorflowgpuproject python = 3.5 tenorflow-gpu = 1.10.0 ipython = 6.5.0 keras matplotlib jupyter scikit-learn numpy = 1.14.5 setuptools = 39.1.0 prompt_toolkit = 1.0.15

conda активировать тензорный потокgpuproject

установка python -m pip --upgradepip --user

setx path "% path%; C: \ Users \ n-bou \ AppData \ Roaming \ Python \ Python35 \ Scripts"

pip uninstall prompt_toolkit (последняя команда быланеобходимо, потому что произошел конфликт между двумя версиями prompt_toolkit)

Затем я смог выполнить свою записную книжку на python, содержащую код UNet, используя керасы с обычным «import тензорным потоком как tf» (вызывая tennorflow-gpu)в начале тетради.

...