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)文章
pytorch 實現(xiàn)在一個優(yōu)化器中設(shè)置多個網(wǎng)絡(luò)參數(shù)的例子
今天小編就為大家分享一篇pytorch 實現(xiàn)在一個優(yōu)化器中設(shè)置多個網(wǎng)絡(luò)參數(shù)的例子，具有很好參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
這篇文章主要給大家介紹了關(guān)于利用Python暴力破解zip文件口令的相關(guān)資料，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧。2017-12-12
PySimpleGUI就是GUI工具包，用于創(chuàng)建GUI的。本文將利用PySimpleGUI編寫一個GUI翻譯器，感興趣的小伙伴可以跟隨小編一起動手嘗試一下2022-08-08
這篇文章主要為大家詳細介紹了chrome配置selenium操作的方法，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-03-03
PyQt為事件處理提供了兩種機制高級的信號與槽機制，以及低級的事件處理機制，這篇文章主要介紹了PyQt5?事件處理機制,需要的朋友可以參考下2022-11-11
這篇文章主要介紹了關(guān)于Python的Thread線程模塊詳解,進程是程序的一次執(zhí)行，每個進程都有自己的地址空間、內(nèi)存、數(shù)據(jù)棧以及其他記錄其運行的輔助數(shù)據(jù),需要的朋友可以參考下2023-05-05
詳解Django+uwsgi+Nginx上線最佳實戰(zhàn)
這篇文章主要介紹了Django+uwsgi+Nginx上線最佳實戰(zhàn)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-03-03
這篇文章主要介紹了如何對numpy?矩陣進行通道間求均值，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
這篇文章主要介紹了在Python里的dict和set的背后小秘密,dict字典是Python中的重要基礎(chǔ)知識,set與其類似,需要的朋友可以參考下2021-11-11
使用keras根據(jù)層名稱來初始化網(wǎng)絡(luò)
這篇文章主要介紹了使用keras根據(jù)層名稱來初始化網(wǎng)絡(luò)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05