import torch

# set device to mps if avaliable, cude if avaliable, cpu otherwise
device = torch.device("mps" if torch.backends.mps.is_available() else "cuda:0" if torch.cuda.is_available() else "cpu")
print (device)

x = torch.zeros(1, device=device)
print (str(x))