1. 程式人生 > 程式設計 >PyTorch使用cpu載入模型運算方式

PyTorch使用cpu載入模型運算方式

沒gpu沒cuda支援的時候載入模型到cpu上計算

model = torch.load(path,map_location=lambda storage,loc: storage.cuda(device))

改為

model = torch.load(path,map_location='cpu')

然後刪掉所有變數後面的.cuda()方法

以上這篇PyTorch使用cpu載入模型運算方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。