Multi GPU train on the ScanNet v2 dataset

[kunzaiyy]

When i use these code to train this network with multi GPU: if torch.cuda.device_count() > 1: model = torch.nn.DataParallel(model) model = model.cuda() It send me this error： RuntimeError: Caught RuntimeError in replica 0 on device 0. Is there any one who can help me? Thank you very much!!!!!!!!!!!1

[GoodJuicee]

When i use these code to train this network with multi GPU: if torch.cuda.device_count() > 1: model = torch.nn.DataParallel(model) model = model.cuda() It send me this error： RuntimeError: Caught RuntimeError in replica 0 on device 0. Is there any one who can help me? Thank you very much!!!!!!!!!!!1

i have the same error , did you solve it? thanks!

[GoodJuicee]

When i use these code to train this network with multi GPU: if torch.cuda.device_count() > 1: model = torch.nn.DataParallel(model) model = model.cuda() It send me this error： RuntimeError: Caught RuntimeError in replica 0 on device 0. Is there any one who can help me? Thank you very much!!!!!!!!!!!1

Hello，did you solve this problem