@G.M I’ve also put the data on GPU via data.to('cuda'). I’ve never had a problem from not specifying the device and just using 'cuda'. Double checking right now data is a PyTorch tensor with device='cuda:0'.