最近在編寫(xiě)深度學(xué)習(xí)的相關(guān)代碼，基于PyTorch，運(yùn)行程序的時(shí)候，報(bào)錯(cuò)，報(bào)錯(cuò)內(nèi)容如下所示：

RuntimeError: expected device cpu but got device cuda:0

歸根結(jié)底，這個(gè)錯(cuò)誤出現(xiàn)的主要原因是GPU類(lèi)型的數(shù)據(jù)和CPU類(lèi)型的數(shù)據(jù)不匹配。了解了錯(cuò)誤的本質(zhì)原因，那如何解決就非常簡(jiǎn)單了。
首先，需要分析大環(huán)境是在GPU上還是在CPU上，然后在出錯(cuò)的位置更改。無(wú)非就兩種情況，分別如下：

• 1.將數(shù)據(jù)改為GPU支持的類(lèi)型

data.cuda()

• 2.將數(shù)據(jù)改為CPU支持的類(lèi)型

data.cpu()

另外，需要注意的一點(diǎn)是：

• 在CUDA下訓(xùn)練中的數(shù)據(jù)不能直接轉(zhuǎn)換為numpy

需要用這行代碼先將cuda數(shù)據(jù)轉(zhuǎn)為cpu數(shù)據(jù)，然后在轉(zhuǎn)為numpy數(shù)據(jù)。

data.cpu().detach().numpy()

這里，用我自己項(xiàng)目里的代碼舉例（環(huán)境是GPU），output是以訓(xùn)練好的網(wǎng)絡(luò)模型對(duì)測(cè)試集某張圖片的預(yù)測(cè)結(jié)果，是Tensor類(lèi)型，支持cuda類(lèi)型的。

總結(jié)

以上是生活随笔為你收集整理的异常解决（一）-- RuntimeError: expected device cpu but got device cuda:0的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

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