PyTorch-YOLOv3 RuntimeError: binary_op(): expected both inputs to be on same device, but input a is on cuda:0 and input b is on cuda:1

RuntimeError: binary_op(): expected both inputs to be on same device, but input a is on cuda:0 and input b is on cuda:1

Open ZhxJia opened this issue 5 years ago • 2 comments

When i want to train the model on multiple gpus, this error appears.So I can only train it on gpu:0

Aug 02 '19 02:08 ZhxJia

Have you solved the problem？I had the same problem as you。

Jun 02 '20 18:06 QinWinner

Is this issue still relevant/occurring?

Sep 14 '21 11:09 Flova