Я пытаюсь использовать 4 gpus в моем labtop. поэтому я использую команду
new_model = multi_gpu_model (model, gpus = 4)
Это работает очень хорошо, но есть одна проблема. только первый GPU имеет больше памяти GPU. ниже мой код и состояние моей памяти gpus проверено watch nvidia-smi
Первый GPU0 имеетувеличиваться только при создании модели до команды new_model = multi_gpu_model(model, gpus=4)
в это время, память 4000 МБ возникает в первом gpu0.
Почему происходит это непревзойденное использование gpu? и как я могу использовать 4 gpus одинаково? пожалуйста, дайте мне подсказку.
большое спасибо.