humanseg.pytorch icon indicating copy to clipboard operation
humanseg.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon cavalleria

multigpu running has bug?

Open ghoshaw opened this issue 3 years ago • 0 comments

Hi, when I run the train code in multigpu, it says that the data is not on the same device. RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 1 does not equal 0 (while checking arguments for cudnn_convolution)

ghoshaw avatar Jun 09 '21 03:06 ghoshaw