(1)第一种GPU训练方式 (二)第二种GPU训练方式 device=torch.device("cpu") device=torch.device("cuda") tudui=tudui.to(device)