Error when running on 2 GPUs

princeton-vl / RAFT

BSD 3-Clause "New" or "Revised" License

3.23k stars 629 forks source link

Error when running on 2 GPUs #179

Open anureetchhabra opened 8 months ago

anureetchhabra commented 8 months ago

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1! (when checking argument for argument mat1 in method wrapper_CUDA_addmm