RuntimeError: Caught RuntimeError in replica 0 on device 0.

jia-zhuang / pytorch-multi-gpu-training

整理 pytorch 单机多 GPU 训练方法与原理

706 stars 80 forks source link

RuntimeError: Caught RuntimeError in replica 0 on device 0. #8

Open LLL-YUE opened 2 weeks ago

LLL-YUE commented 2 weeks ago

When I use torch.nn.DataParallel, the error will appear: RuntimeError: Caught RuntimeError in replica 0 on device 0. How can I deal with this situation?