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