Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

Question

Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

katariaak579 opened this issue a year ago · 1 comments

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument grid in method wrapper_CUDA__grid_sampler_2d) whilke running infer.py

Answer 1 · 2024-12-18T08:46:19.000Z

using python3 infer.py --deca DECA --rome data --device cuda