PyTorch改为默认使用GPU

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