pytorch设置gpu 方法一: import torch device = torch.device("cuda:3" if torch.cuda.is_available() else "cpu") model = model.to(device) 方法二: torch.cuda.set_device(0) model = Network() model = model.cuda()