PyTorch學習筆記(3)—CPU和GPU上加載模型

前言 有一些現實的問題是這樣的：當我們在GPU集羣或者服務器上訓練模型的時候，有時候需要將模型取回，在本地測試一下。這個時候就需要PyTorch將模型轉換爲cpu的版本，因爲PyTorch針對不同的系統和cuda有不同的版本。因此無法直接將GPU訓練出的模型直接用於CPU來做inference，需要進行一些簡單的變化。 轉換（轉自牧野 ——pytorch在CPU和GPU上加載模型） pytorch

>>阅读原文<<