Установить Keras для GPU?

Question

Я установил Tensorflow для графического процессора, используя: pip install tensorflow-gpu Но когда я попробовал то же самое для Керас : pip install keras-gpu, я получил ошибку: не смог найти версию, которая удовлетворяет требованиям

Надеюсь, кто-то найдет и поможет мне!Спасибо !

Answer 1

Для установки tenorflow-gpu с определенной версией cuda 9.0 используйте:

"conda install tenorflow-gpu cudatoolkit = 9.0 -c anaconda"

Аналогично для keras-gpu

Answer 2

Добавление к ответу выше, который является правильным с точки зрения рекомендации по использованию менеджера пакетов Anaconda, но устарел в том, что теперь содержит пакет keras-gpu на Anaconda Cloud .

Итак, после установки Anaconda вам просто нужно создать новую среду, в которой вы хотите установить keras-gpu, и выполнить команду:

conda install -c anaconda keras-gpu

Это установит Keras вместе с библиотеками тензор-поток и тензор-поток-gpu в качестве бэкэнда.(Также нет необходимости устанавливать отдельно библиотеки времени выполнения CUDA и библиотеки cudnn, поскольку они также включены в пакет - протестировано в Windows 10 и работает).

Answer 3

В дополнение к двум приведенным выше ответам убедитесь, что ваша среда TensorFlow / Keras использует Python 3.6.Keras / TensorFlow не очень хорошо работает с Python 3.7, по состоянию на 10 мая 2019 года.

Я попытался использовать Keras / TensorFlow с Python 3.7, и мне пришлось переустановить Anaconda, так как он сломалсяМоя Анаконда Подскажите.

Answer 4

Существует , а не любой keras-gpu пакет [ ОБНОВЛЕНИЕ : сейчас есть, см. Другой ответ ниже];Keras - оболочка для некоторых бэкэндов, включая Tensorflow, и эти бэкэнды могут быть разных версий, например tensorflow и tensorflow-gpu.Но это не относится к самой Keras, которая должна быть установлена ​​просто с

pip install keras

независимо от того, какой бэкэнд используется (см. PyPi docs ).

Дополнительнои поскольку вы пометили вопрос как anaconda, также имейте в виду, что, как правило, не рекомендуется смешивать менеджеры пакетов (т. е. pip с conda), и вам может быть лучше установить Keras из Anaconda cloud с

conda install -c conda-forge keras 

Наконец, вам также может быть интересно узнать, что последние версии Tensorflow включают Keras в качестве подпакета, так что вы можете использовать его без какой-либо дополнительной установки;см. https://www.tensorflow.org/guide/keras

Для установки tensorflow-gpu из Anaconda cloud , вам необходимо использовать

conda install -c anaconda tensorflow-gpu 

перед установкой Keras.Убедитесь, что вы делаете это в другой виртуальной среде или после удаления других версий (т.е. pip -установленных), так как сообщалось о проблемах в противном случае.