Torch.cuda.set Device 0 1 2 3