Hi,
Saving Tensors does save their device as it is a property of the object.
You can use the map_location argument from torch.load 46 to avoid any issue if the device it was saved for is not available anymore. In this case, you want to use "cuda:0" I guess for the new location).