解決pytorch報錯:AssertionError: Invalid device id的問題
在伺服器上訓練的網路放到本地臺式機進行infer，結果出現報錯：
AssertionError: Invalid device id
仔細檢查後發現原來伺服器有多個GPU，當時開啟了兩個進行加速運算。
net1 = nn.DataParallel(net1,device_ids=[0,1])
而本地臺式機只有一個GPU，呼叫數量超出所以報錯。
改為
net1 = nn.DataParallel(net1,device_ids=[0])
問題解決。
以上這篇解決pytorch報錯:AssertionError: Invalid device id的問題就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。
相關推薦
解決pytorch報錯:AssertionError: Invalid device id的問題
在伺服器上訓練的網路放到本地臺式機進行infer，結果出現報錯： AssertionError: Invalid device id
SpringBoot Undertow/Tomcat報錯：Invalid character | in request-target
本問題核心原因是http協議升級，因為網路安全問題日益嚴峻，RFC組織決定將RFC2616拆分並進行升級。
Android Studio 解決編譯報錯 Could not download aapt2-windows.jar
在進行編譯apk時，出現以下報錯: Could not download aapt2-windows.jar (com.android.tools.build:aapt2:3.4.1-5326820): No cached version available for offline mode
解決12899報錯
出現ORA-12899,是字符集引起的,中文在UTF-8中佔3個位元組,ZHS16GBK中佔2個位元組,而源dmp檔案字符集是ZHS16GBK庫裡倒出來的資料,現在要匯入到目標字符集為UTF-8的庫裡,所以會出現ORA-12899
解決android報錯:Intel HAXM is required to run this AVD
今天，簡單講解Android 啟動模擬器時，提示錯誤： Intel HAXM is required to run this AVD. VT-x is disabled in BIOS. Enable VT-x in your BIOS security settings (refer to documentation for your computer)的
解決“tensorboard報錯”問題
技術標籤：踩坑記錄linuxbugpytorch 問題描述 在執行程式程式碼的時候，遇到如下報錯資訊：
解決urllib2報錯
技術標籤：pythonpythonpycharmbug 解決urllib2報錯 報錯資訊如下解決方案 報錯資訊如下
else報錯SyntaxError: invalid syntax
技術標籤：錯誤記錄python 原始碼： def leave(event): # <Esc>事件處理程式 ret = messagebox.askyesno("ch11_4", "是否離開?")
【問題解決】報錯：ImportError: No module named _libfdt
技術標籤：# 全志Linux 問題： 編譯uboot的時候報錯： Traceback (most recent call last): File "./tools/binman/binman", line 32, in <module>
[pytorch][zero_grad]pytorch報錯‘builtin_function_or_method‘ object has no attribute ‘named_modules‘
技術標籤：pytorchpytorchpythondebug神經網路 pytorch做grad-cam模型梯度歸零時呼叫了zero_grad這個函式，然後報錯
vue如何引入本地json，並解決圖片報錯問題
技術標籤：圖片處理vuejsonjavascriptjsimg 今天小編引入json假資料，引入圖片載入不出來，特此寫了此文記錄一下，方便你我他 本地json引入方式
js前端處理過濾特殊字元以及輸入法特殊表情符號emoji的正則方法，解決資料庫報錯問題。
技術標籤：工作分享jqueryhtmljavascript 現在，輸入法基本上都支援表情輸入和特殊字元，並且可以在input框內輸入表情， 如果我們不對這些表情進行驗證的話，我們會發現，有的時候也可以正常 進行儲存，但是這樣
Linux docker 刪除映象檔案 以及解決刪除報錯image is being used by stopped container 的問題
技術標籤：Linuxlinuxdocker 1.使用命令檢視所有映象 docker images 2.根據ID刪除需要刪除的映象
ubuntu安裝mysql遇到的坑----解決Mysql報錯缺少libaio.so.1
最近學習大資料，涉及到hive的部分需要安裝mysql，於是就在linux環境下嘗試安裝，對於我這個linux小白來說，中間遇到很多坑爹問題，在這裡做一個記錄。
解決Django報錯:SQLITE 3.8.3 OR LATER IS REQUIRED【轉】
django在linux伺服器執行python3 manage.py migrate，會產生了如下的錯誤輸出： django.core.exceptions.ImproperlyConfigured: SQLite 3.8.3 or later is required (found 3.7.17).
解決elasticsearch報錯報錯“java.lang.IllegalArgumentException: Rejecting mapping update to [這裡是索引名稱保密] as the final mapping would have more than 1 type: [_doc, log]"”
elasticsearch index索引 某日在研究kibana的索引生命週期功能，感覺對於我們現在幾千個索引蠻有用途，之前都是寫個刪除指令碼呢，放到定時任務進行刪除。
[PHP] 解決laravel 報錯：Too Many Attempts
當使用ab測試工具測試介面的時候 在日誌裡看到了很多Too Many Attempts的錯誤 這是因為laravel的限流中介軟體在起作用
如何解決 MySQL報錯：ERROR 1045 (28000)
如何解決 MySQL報錯：ERROR 1045 (28000) 一、問題復現 ┌──(rabbit㉿ASUS)-[/mnt/c/Users/rabbit]
解決python報錯：ModuleNotFoundError: No module named '_sysconfigdata_x86_64_conda_linux_gnu'
在近期conda的版本更新中，有可能會刪除路徑下的_sysconfigdata_x86_64_conda_linux_gnu模組相關備份檔案，而在其他的一些軟體的執行過程中有可能依賴於這個備份檔案，這就會導致執行過程報錯模組無法找到的問題。
[IDEA]中 javaweb的web.xml檔案報錯：Invalid content was found starting with element等錯誤
報錯現場還原 web.xml檔案頭部宣告如下： <?xml version=\"1.0\" encoding=\"UTF-8\"?> <web-app xmlns=\"http://xmlns.jcp.org/xml/ns/javaee\"