解決pytorch報(bào)錯(cuò):AssertionError: Invalid device id的問題
更新時(shí)間：2020年01月10日 09:06:37 作者：學(xué)EE的粒粒
今天小編就為大家分享一篇解決pytorch報(bào)錯(cuò):AssertionError: Invalid device id的問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
在服務(wù)器上訓(xùn)練的網(wǎng)絡(luò)放到本地臺(tái)式機(jī)進(jìn)行infer，結(jié)果出現(xiàn)報(bào)錯(cuò)：
AssertionError: Invalid device id
仔細(xì)檢查后發(fā)現(xiàn)原來服務(wù)器有多個(gè)GPU，當(dāng)時(shí)開啟了兩個(gè)進(jìn)行加速運(yùn)算。
net1 = nn.DataParallel(net1, device_ids=[0, 1])
而本地臺(tái)式機(jī)只有一個(gè)GPU，調(diào)用數(shù)量超出所以報(bào)錯(cuò)。
改為
net1 = nn.DataParallel(net1, device_ids=[0])
問題解決。
以上這篇解決pytorch報(bào)錯(cuò):AssertionError: Invalid device id的問題就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python實(shí)現(xiàn)的HTTP并發(fā)測(cè)試完整示例
這篇文章主要介紹了Python實(shí)現(xiàn)的HTTP并發(fā)測(cè)試,涉及Python多線程并發(fā)操作相關(guān)實(shí)現(xiàn)技巧,需要的朋友可以參考下2015-05-05
Django 接收Post請(qǐng)求數(shù)據(jù),并保存到數(shù)據(jù)庫的實(shí)現(xiàn)方法
今天小編就為大家分享一篇Django 接收Post請(qǐng)求數(shù)據(jù),并保存到數(shù)據(jù)庫的實(shí)現(xiàn)方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-07-07
大家好，本篇文章主要講的是用Python替換證件照背景顏色，感興趣的同學(xué)趕快來看一看吧，對(duì)你有幫助的話記得收藏一下2022-01-01
這篇文章主要介紹了python黑魔法之參數(shù)傳遞，分析了python參數(shù)傳遞的方法，感興趣的小伙伴們可以參考一下2016-02-02
解決windows下命令行執(zhí)行python3失效,會(huì)打開應(yīng)用商店問題
這篇文章主要介紹了解決windows下命令行執(zhí)行python3失效,會(huì)打開應(yīng)用商店問題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-02-02
這篇文章主要介紹了Django 刪除upload_to文件的步驟，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-03-03
今天小編就為大家分享一篇pandas dataframe添加表格框線輸出的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-02-02
跳出單循環(huán)不管是什么編程語言,都有可能會(huì)有跳出循環(huán)的需求,本文主要介紹了Python中斷多重循環(huán)的幾種方法,具有一定的參考價(jià)值,感興趣的可以了解一下2023-11-11
JS設(shè)計(jì)模式之責(zé)任鏈模式實(shí)例詳解
這篇文章主要介紹了JS設(shè)計(jì)模式之責(zé)任鏈模式,結(jié)合實(shí)例形式詳細(xì)分析了責(zé)任鏈模式的概念、原理、功能、使用場(chǎng)景及相關(guān)操作技巧,需要的朋友可以參考下2018-02-02
在本篇文章中小編給大家分享的是關(guān)于pycharm重命名文件的方法步驟，對(duì)此有需要的朋友們可以學(xué)習(xí)參考下。2019-07-07