pytorch学习笔记（6）：GPU和如何保存加载模型

参考文档：https://mp.weixin.qq.com/s/kmed_E4MaDwN-oIqDh8-tg

上篇文章我们完成了一个 vgg 网络的实现，那么现在已经掌握了一些基础的网络结构的实现，距离一个入门炼丹师还有两个小问题需要注意一下：GPU 和保存模型。

提起炼丹大家经常可以听到显卡如何如何的，也就是 GPU 在炼丹的过程中起到重要的作用。另一方面，训练了一个模型后，我们肯定要用它来进行一些预测，前面的代码中都是将训练好的模型直接进行预测，但是如果代码每次预测都要训练一次岂不是麻烦死了，所以将训练好的模型保存下来也是一个关键环节。

1、GPU在炼丹中如何使用

GPU 擅长并行式的图像计算，而张量本身和图像一样都是矩阵计算，所以对于 tensor 的计算，GPU 本身就有得天独厚的优势。再加上并行式的训练方式，可以有效的节约训练时候的时间消耗。
那么我们怎么在训练过程中使用 GPU 呢？

首先安装 pytorch 的时候在官网需要选择 cuda 的版本，这里我们不多赘述。然后通过下面的代码查看你的电脑是否支持 GPU，以及是否安装成功。

torch.cuda.is_available()

如果显示为