Pytorch——报错解决：RuntimeError: Expected all tensors to be on the same device, but found at least two_when checking argument for argument target in meth

作者：盐析白兔 | 2024-02-27 02:59:37

踩

when checking argument for argument target in method wrapper_nll_loss_forwar

Pytorch——报错解决：RuntimeError: Expected all tensors to be on the same device, but found at least two devices

一、报错信息
二、报错原因
三、解决方法
四、原代码
五、现代码
参考博客

一、报错信息

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument weight in method wrapper_nll_loss_forward)
在这里插入图片描述

二、报错原因

tensors 一会在CPU运算，一会在GPU运算。

三、解决方法

都改成在GPU运算。

四、原代码

在这里插入图片描述

五、现代码

在这里插入图片描述

参考博客

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/149492