И в догонку: Если сделать model = nn.DataParallel(model), то моделька будет

Question

выполняться сразу же на всех доступных GPU, а вот будет ли в этом случае и выгрузка данных с их обработкой выполняться на CPU или тоже зацепит GPU?

0

31.03.2018

11 просмотров

Evgenii Zheltonozhskii🇮🇱 · Accepted Answer

выгрузка на гпу не может быть выполнена гпу по понятным причинам

0

31.03.2018