PyTorch学习笔记(3)—CPU和GPU上加载模型

前言

有一些现实的问题是这样的：当我们在GPU集群或者服务器上训练模型的时候，有时候需要将模型取回，在本地测试一下。这个时候就需要PyTorch将模型转换为cpu的版本，因为PyTorch针对不同的系统和cuda有不同的版本。因此无法直接将GPU训练出的模型直接用于CPU来做inference，需要进行一些简单的变化。

转换（转自牧野 ——pytorch在CPU和GPU上加载模型）

pytorch允许把在GPU上训练的模型加载到CPU上，也允许把在CPU上训练的模型加载到GPU上。