a problem when train

Open pokameng opened this issue 4 years ago • 3 comments

RuntimeError: A tensor was not on the same device as the first tensor

When i run the train.py ,i face up with this problem

Apr 27 '21 08:04 pokameng

Are you using multiple GPUs for training?

May 10 '21 23:05 DevTech15

Yeah 4 gpus

---Original--- From: @.> Date: Tue, May 11, 2021 07:29 AM To: @.>; Cc: @.@.>; Subject: Re: [NVlabs/STEP] a problem when train (#21)

Are you using multiple GPUs for training?

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

May 11 '21 00:05 pokameng

Please try it on single GPU for now. Then we can try to train it on multiple GPUs.

May 12 '21 00:05 DevTech15