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

t192306
Mon, 15 May 2023 13:05:48 +0000

@pchhapolika - I am using a 8 GPU Cluster and I am encountering the same issue. Did you happen to get any solution yet? Thanks!