11/27日学习

#P30-P33完结

 

利用GPU来训练

方法1：

思路：找到网络模型、数据（输入，标注）、损失函数。然后调用.cuda() 然后返回

 

方法2：

Device=torch.device("CPU")

torch.device("cuda:0")　　#指定显卡

.to(device)