Как сделать несколько GPU-тренировок с Keras?

Я хочу, чтобы моя модель запускалась на нескольких параметрах совместного использования графических процессоров, но с разными партиями данных.

Могу ли я сделать что-то подобное с помощью model.fit()? Есть ли другая альтернатива?

6
18 июля '17 в 15:02
источник поделиться
2 ответов

попробуйте использовать функцию make_parallel в: https://github.com/kuza55/keras-extras/blob/master/utils/multi_gpu.py (он будет работать только с бэкэнд-тензором).

1
26 сент. '17 в 18:29
источник

У Keras теперь (по версии 2.0.9) встроенная поддержка устройства parallelism на нескольких графических процессорах, используя keras.utils.multi_gpu_model.

В настоящее время поддерживается только поддержка Tensorflow.

Хороший пример здесь (docs): https://keras.io/getting-started/faq/#how-can-i-run-a-keras-model-on-multiple-gpus Также здесь: https://datascience.stackexchange.com/a/25737

0
31 дек. '17 в 21:52
источник

Посмотрите другие вопросы по меткам или Задайте вопрос