最近在運行pytorch代碼時發現的該錯誤，不管是 expected cpu 還是 expected cuda， 本質原因都是類型不匹配。

一般是因為：

等號左邊和右邊類型不一樣

運算符左右兩端類型不同，例：+ - * /

同一個函數內，傳入參數的類型不同，例matmul等

你要考慮你的大環境是在什么上面部署的,CPU 還是 CUDA,然后再出錯位置嘗試修改數據.

->cuda : data.cuda()

->cpu: data.cpu()

->numpy：注意cuda類型不能直接轉numpy 須先轉成Cpu類型，data.cpu().numpy()? ?

? ? ? ? ? ? ? ? ? 注意在CUDA下訓練中的數據不能直接轉換為numpy，data.cpu().detach().numpy()

總結

以上是生活随笔為你收集整理的运行时报错RuntimeError: expected device cpu but got device cuda:0的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。