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è)參考,也希望大家多多支持腳本之家。
您可能感興趣的文章:
- python神經(jīng)網(wǎng)絡(luò)pytorch中BN運(yùn)算操作自實(shí)現(xiàn)
- python神經(jīng)網(wǎng)絡(luò)學(xué)習(xí)利用PyTorch進(jìn)行回歸運(yùn)算
- pytorch中常用的乘法運(yùn)算及相關(guān)的運(yùn)算符(@和*)
- PyTorch中Tensor的數(shù)據(jù)類型和運(yùn)算的使用
- Pytorch Tensor基本數(shù)學(xué)運(yùn)算詳解
- Pytorch 多維數(shù)組運(yùn)算過(guò)程的索引處理方式
- pytorch torch運(yùn)算操作示例詳解
相關(guān)文章
pytorch 實(shí)現(xiàn)在一個(gè)優(yōu)化器中設(shè)置多個(gè)網(wǎng)絡(luò)參數(shù)的例子
今天小編就為大家分享一篇pytorch 實(shí)現(xiàn)在一個(gè)優(yōu)化器中設(shè)置多個(gè)網(wǎng)絡(luò)參數(shù)的例子,具有很好參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-02-02
詳解Django+uwsgi+Nginx上線最佳實(shí)戰(zhàn)
這篇文章主要介紹了Django+uwsgi+Nginx上線最佳實(shí)戰(zhàn),文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-03-03
使用keras根據(jù)層名稱來(lái)初始化網(wǎng)絡(luò)
這篇文章主要介紹了使用keras根據(jù)層名稱來(lái)初始化網(wǎng)絡(luò),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-05-05

