原文:【深度學習】RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

報錯代碼: 報錯截圖如下: 報這個錯的原因在於,代碼中的Tensor,一會在CPU中運行,一會在GPU中運行,所以最好是都放在同一個device中執行 核心代碼: 並且將用到的Tensor都改為同一個device:Tensor.to device 上述代碼修改后: 這樣就不會報錯了 完整代碼: ...

2022-03-18 10:49 0 5805 推薦指數:

查看詳情

張量與數組運算報錯(Use Tensor.cpu() to copy the tensor to host memory first;RuntimeError: Expected all tensors to be on the same device)

結論:1.張量與數組運算,張量必須在cpu上,產生結果為cpu上的張量,可繼續與數組運算(張量必須在gpu上)    2.張量與張量運算,cpu上的張量與gpu上的張量是無法運行的,必須在相同的gpu上或cpu上,猜想不同型號的gpu因該也不行。 一.張量與數組運算,前提張量 ...

Tue Mar 16 09:04:00 CST 2021 0 6880
【E-14-Pytorch】RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat1' in call to _th_addmm

一、出錯誤背景: Pytorch 中想使用 CUDA 對程序計算進行加速 二、問題分析 錯誤的意思:object 的 device 類型期望得到的是 cuda 類型,但是實際上的類型確實 cpu 類型,在調用二分類交叉熵損失進行前向計算的時候 三、檢查下面幾點: 模型是否放到 ...

Sat Oct 24 02:37:00 CST 2020 0 4751
RuntimeError: CUDA error: invalid device ordinal

報錯內容: RuntimeError: CUDA error: invalid device ordinal 雖然這里沒顯示,但是切換CUDA,可以看到確實在使用英偉達顯卡。 ...

Mon Apr 13 02:18:00 CST 2020 1 7458
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM