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