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