PyTorch使用cpu加載模型運算方式
更新時間：2020年01月13日 09:51:02 作者：ngy321
今天小編就為大家分享一篇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加載模型運算方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了最大K個數(shù)問題的Python版解法總結(jié),以最大K個數(shù)問題為基礎(chǔ)的算法題目在面試和各大考試及競賽中經(jīng)常出現(xiàn),需要的朋友可以參考下2016-06-06
本篇文章給大家分享了Python應(yīng)用庫大全的相關(guān)內(nèi)容，對此有需要的可以學(xué)習(xí)下。2018-05-05
這篇文章主要介紹了django列表篩選功能的實現(xiàn)代碼，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03
下面小編就為大家分享一篇Python之ReportLab繪制條形碼和二維碼的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-01-01
python實現(xiàn)ip地址查詢經(jīng)緯度定位詳解
這篇文章主要介紹了python實現(xiàn)ip地址查詢經(jīng)緯度定位詳解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-08-08
python實現(xiàn)圖片轉(zhuǎn)換成素描和漫畫格式
這篇文章主要為大家詳細(xì)介紹了python實現(xiàn)圖片轉(zhuǎn)換成素描和漫畫格式，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2020-08-08
搭建Python的Django框架環(huán)境并建立和運行第一個App的教程
時至今日Django仍然是人氣最高的Web開發(fā)框架,MVC結(jié)構(gòu)清晰內(nèi)置功能強大,這里我們就來共同學(xué)習(xí)搭建Python的Django框架環(huán)境并建立和運行第一個App的教程2016-07-07
Python lambda表達(dá)式filter、map、reduce函數(shù)用法解析
這篇文章主要介紹了Python lambda表達(dá)式filter、map、reduce函數(shù)用法解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-09-09
Python爬蟲獲取國外大橋排行榜數(shù)據(jù)清單
這篇文章主要介紹了Python爬蟲獲取國外大橋排行榜數(shù)據(jù)清單，文章通過PyQuery?解析框架展開全文詳細(xì)內(nèi)容，需要的小伙伴可以參考一下2022-05-05
python+html實現(xiàn)前后端數(shù)據(jù)交互界面顯示的全過程
最近項目中采用了前后端分離的技術(shù),感覺有必要給大家總結(jié)下,所以下面這篇文章主要給大家介紹了關(guān)于python+html實現(xiàn)前后端數(shù)據(jù)交互界面顯示的相關(guān)資料,需要的朋友可以參考下2022-06-06