最近在運行pytorch代碼時發(fā)現(xiàn)的該錯誤，不管是 expected cpu 還是 expected cuda， 本質(zhì)原因都是類型不匹配。

一般是因為：

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

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

同一個函數(shù)內(nèi)，傳入?yún)?shù)的類型不同，例matmul等

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

->cuda : data.cuda()

->cpu: data.cpu()

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

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。