RuntimeError about pytorch1.7.1

RuntimeError: iter.device(arg).is_cuda() INTERNAL ASSERT FAILED at “/pytorch/aten/src/ATen/native/cuda/Loops.cuh”:94

Can you check if the inputs to the computation are all on the same device type (and if they are on GPU, the same GPU)?