解決pytorch報錯:AssertionError: Invalid device id的問題
更新時間：2020年01月10日 09:06:37 作者：學(xué)EE的粒粒
今天小編就為大家分享一篇解決pytorch報錯:AssertionError: Invalid device id的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
在服務(wù)器上訓(xùn)練的網(wǎng)絡(luò)放到本地臺式機進行infer，結(jié)果出現(xiàn)報錯：
AssertionError: Invalid device id
仔細檢查后發(fā)現(xiàn)原來服務(wù)器有多個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)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要為大家介紹了python函數(shù)默認參數(shù)使用的踩雷避坑詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-07-07
這篇文章主要介紹了Django配置跨域并開發(fā)測試接口,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-11-11
這篇文章主要介紹了Python yield 使用淺析,本文給出了多個使用實例來分析yield的使用方法,需要的朋友可以參考下2015-05-05
基于Django框架利用Ajax實現(xiàn)點贊功能實例代碼
點贊這個功能是我們現(xiàn)在經(jīng)常會遇到的一個功能，下面這篇文章主要給大家介紹了關(guān)于基于Django框架利用Ajax實現(xiàn)點贊功能的相關(guān)資料，文中通過示例代碼介紹的非常詳細，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2018-08-08
這篇文章主要介紹了python requests庫的使用，幫助大家更好的利用python進行爬蟲，感興趣的朋友可以了解下2021-01-01
這篇文章主要介紹了Python字符串的轉(zhuǎn)義字符，轉(zhuǎn)義字符是指，用一些普通字符的組合來代替一些特殊字符，由于其組合改變了原來字符表示的含義，下文相關(guān)資料需要的小伙伴可以參考一下2022-04-04
這篇文章主要介紹了Python迭代器和生成器定義與用法,結(jié)合實例形式詳細分析了Python迭代器和生成器的概念、原理、定義、使用方法及相關(guān)操作注意事項,需要的朋友可以參考下2018-02-02
Python開發(fā)生產(chǎn)環(huán)境常用的4個工具(實用推薦)
構(gòu)建優(yōu)秀的軟件需要遵循特定的規(guī)則并執(zhí)行行業(yè)標準,如何在真實的生產(chǎn)環(huán)境開發(fā)中體現(xiàn)呢？在這篇文章中,我將向您展示我在Python項目中設(shè)置的4種工具,以簡化開發(fā)工作流程并執(zhí)行一些最佳實踐,這些工具幫助我提高了效率,節(jié)省了時間,希望你讀完也能有所收獲2024-01-01
這篇文章主要介紹了詳解將Django部署到Centos7全攻略，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-09-09
本文主要介紹了Python中查找缺失值的三種方法,包括pandas庫的isnull()方法、numpy庫的isnan()方法和scikit-learn庫的SimpleImputer類,感興趣的可以了解一下2023-11-11