#P30-P33完结
利用GPU来训练
方法1:
思路:找到网络模型、数据(输入,标注)、损失函数。然后调用.cuda() 然后返回
方法2:
Device=torch.device("CPU")
torch.device("cuda:0") #指定显卡
.to(device)