PyTorch默认用CPU,改默认用GPU,
import torch后,用
torch.set_default_device('cuda') 改为默认使用GPU