PyTorch学习笔记（9）使用GPU训练模型并测试自己的模型

对之前的训练再优化，如果自己的电脑没有 GPU，也可以去网上找一些服务，如谷歌、阿里云等。

---

使用GPU训练

只需要将网络模型、损失函数、数据放入 cuda() 即可。

例如：

loss_fn = nn.CrossEntropyLoss()
loss_fn = loss_fn.cuda()

imgs,targets = data
imgs = imgs.cuda()
targets = targets.cuda()

但是没有 GPU 的时候会报异常，所以最好加判断：

loss_fn = nn.CrossEntropyLoss()
if torch.cuda.is_available():
	loss_fn = loss_fn.cuda()

显然，这样写，代码修改起来很麻烦，因此还有更简便的方法。

# 在最开头加上此句
# 定义训练的设备
device = torch.device("cuda" if torch.cuda.is_available() else