style-based-gan-pytorch icon indicating copy to clipboard operation
style-based-gan-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon rosinality

Uses only 1 GPU instead of four

Open Vova-B opened this issue 6 years ago • 1 comments

I run training on four gpu gtx2080ti but training does not start and the error falls Cuda Out of memory GPU: 0, I looked and saw that the program uses only 1 GPU instead of 4, how to fix it?

Vova-B avatar Aug 14 '19 15:08 Vova-B

By default train script will use all available gpus using nn.DataParallel. Could you re-check your gpu settings, for example using nn.DataParallel?

rosinality avatar Aug 14 '19 15:08 rosinality