当前位置：网站首页>Pytorch——报错解决：RuntimeError: Expected all tensors to be on the same device, but found at least two

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

2022-07-16 02:51:00 【William.csj】

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!

原网站

版权声明
本文为[William.csj]所创，转载请带上原文链接，感谢
https://blog.csdn.net/Williamcsj/article/details/125745982