原文:RuntimeError: Expected object of backend CUDA but got backend CPU for argument #4 'mat1'

原因:變量沒有加cuda data data.cuda ...

2019-09-03 17:19 0 4232 推薦指數:

查看詳情

【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
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM