PyTorch: запуск вывода на нескольких графических процессорах ⇐ Python

[Anonymous]

У меня есть модель, которая принимает два входных параметра. Я хочу выполнить вывод на нескольких графических процессорах, где один из входов фиксирован, а другой изменяется. Итак, допустим, я использую n графических процессоров, у каждого из них есть копия модели. Первый графический процессор обрабатывает входную пару (a_1, b), второй обрабатывает (a_2, b) и так далее. Все выходные данные сохраняются в виде файлов, поэтому мне не нужно выполнять операцию объединения выходных данных. Как я могу сделать это с помощью DDP или другим способом?

Подробнее здесь: https://stackoverflow.com/questions/739 ... tiple-gpus