欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

PyTorch使用cpu加載模型運(yùn)算方式

 更新時(shí)間:2020年01月13日 09:51:02   作者:ngy321  
今天小編就為大家分享一篇PyTorch使用cpu加載模型運(yùn)算方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

沒(méi)gpu沒(méi)cuda支持的時(shí)候加載模型到cpu上計(jì)算

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

改為

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

然后刪掉所有變量后面的.cuda()方法

以上這篇PyTorch使用cpu加載模型運(yùn)算方式就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。

相關(guān)文章

最新評(píng)論