Использование памяти Keras Multi GPU отличается - PullRequest
1 голос
/ 16 октября 2019

Я пытаюсь использовать 4 gpus в моем labtop. поэтому я использую команду

new_model = multi_gpu_model (model, gpus = 4)

Это работает очень хорошо, но есть одна проблема. только первый GPU имеет больше памяти GPU. ниже мой код и состояние моей памяти gpus проверено watch nvidia-smi enter image description here

enter image description here

Первый GPU0 имеетувеличиваться только при создании модели до команды new_model = multi_gpu_model(model, gpus=4)

в это время, память 4000 МБ возникает в первом gpu0.

Почему происходит это непревзойденное использование gpu? и как я могу использовать 4 gpus одинаково? пожалуйста, дайте мне подсказку.

большое спасибо.

...