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)

[HongguLiu]

Hello, When I used multiple gpus to train model, I usually get this error. 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) Could you tell me how to sole it.

best, Honggu Liu

[AAA-Fan]

Same issure. Did you fix it?

[surprisedong]

Specify a gpu device using 'CUDA_VISIBLE_DEVICES=0'