RuntimeError: Expected object of device type cuda but got device type cpu for argument #2

最新推荐文章于 2023-12-24 15:58:07 发布

原创最新推荐文章于 2023-12-24 15:58:07 发布 · 2.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

本文记录了一次在PyTorch中遇到的运行时错误，即在调用矩阵乘法时，期望的设备类型为CUDA，但实际传入的是CPU类型。作者通过尝试将自定义函数gc()转换为nn.Module类的方法未果，最终成功地通过确保所有操作都在GPU上执行解决了问题。关键在于将线性层实例化时显式调用.cuda()方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat2' in call to _th_mm

gate = gc(img,img_org,cap)

应该是说我这个地方启用了CPU没有放到GPU里，查了一些方法，mark一下

我这里gc()是一个def，试着把gc()写成nn.Model的类函数（失败）
试着把gc()移到class model()里，失败
最后，成功，我真的是太小白了

before：

self.linear = nn.Linear(output1.size()[0], 1)

now：

self.linear = nn.Linear(output1.size()[0], 1).cuda()

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

AbyssLeo

关注关注

1
点赞
踩
0

收藏

觉得还不错? 一键收藏
2
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat1' in

alittlebai1的博客

04-14

1万+

这是我在第一次用pytorch的时候遇见的一个问题，对于新手来说，折磨我一个多小时。我用的pycharm 解决办法： Google 一下，有这么个博客：检查下面几点: 1.模型是否放到了CUDA上model = model. to(device)或model = model. cuda(device)； 2.输入数据是否放到了CUDA上data = data. to(device)或data ...

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'target'

CICI❤️的博客

11-11

2万+

2 条评论您还未登录，请先登录后发表或查看评论

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 ‘mat2‘ in

wavehaha的博客

04-22

6129

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 ‘mat2’ in call to _th_mm 最近在利用Linux服务器训练强化学习模型，由于是现在CPU上跑的，后来放到了GPU跑，结果出现了上述问题，大致意思是object 的 device 类型期望得到的是 cuda 类型，但是实际上的类型确实 cpu 类型经检查发现虽然device = torch.device(

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 ‘target‘

点赞的都能无bug通过！

12-07

1674

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 ‘target’ in call to _thnn_binary_cross_entropy_forwardPytorch 中想使用 CUDA 对程序计算进行加速object 的 device 类型期望得到的是 cuda 类型，但是实际上的类型确实 cpu 类型，在调用二分类交叉熵损失进行前向计算的时候检查下面几点一般情况下应该是

RuntimeError: Expected object of device type cuda but got device type cpu for argument pytorch数据位置

kyle1314608的博客

06-09

3606

RuntimeError: Expected object of device type cuda but got devic...

RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 ‘self‘ in

最新发布

m0_56393391的博客

12-24

589

RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _thnn_conv2d_forward

2021-11-08RuntimeError: Expected object of device type cuda but got device type cpu for argument #3

weixin_43444314的博客

11-08

810

报错RuntimeError: Expected object of device type cuda but got device type cpu for argument #3 'weight' in call to _thnn_nll_loss_forward报错思路功能快捷键合理的创建标题，有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTe

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat2' in call to _th_bmm_out

07-12

这个错误通常是由于在CUDA设备上执行操作时，输入的张量不在CUDA设备上引起的。根据错误消息，这里的问题是第二个输入张量没有被正确地移动到CUDA设备上。要解决这个问题，你可以使用`.to()`方法将张量移动到正确...

运行时错误：应为设备类型cuda的对象，但在中为参数“mat1”获取了设备类型cpu,RuntimeErrorExpectedobjectofdevicetypecudabutgotdevicetyp

每日提升一点点

08-01

461

https://www.pythonf.cn/read/94009

nn.Embedding踩坑：RuntimeError: Expected object of device type cuda but got device type cpu

SHY0978的博客

03-24

1383

nn.Embedding踩坑

RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 ‘target‘

goodfake

09-20

1246

一般有三点错误 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 1、模型是否放到了CUDA上 model = model.to(device) 或 model = model.cuda() 2、输入数据是否放到了CUDA上 data = data.to(device) 或 data = data.cuda() 3、模型内部新建的张量是否放到了CUDA上 p = torch.tensor([1]).t

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 ‘mat1‘ in

tangweirensheng的博客

10-13

674

pycharm运行的好好的，一上服务器都是问题，哭了 RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat1' in call to _th_addmm Pytorch中想使用CUDA对程序计算进行加速，object的device类型期望得到的是cuda类型，但是实际上的类型确实cpu类型 device = torch.device("cuda...

Expected object of device type cuda but got device type cpu...

03-21

2907

Pytorch RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 ‘self’ in call to _th_index_select 这个报错的原因是：模型的部分参数或数据不在GPU而在CPU上。遇到这个问题需要仔细排查，　模型是否已经放在在GPU上，　通常使用 model = model.cuda() data = data.cuda() 把模型和数据分别放到GPU上。

pytorch转onnx模型常见的Expected object of device type cuda but got device type cpu错误分析

lhyyhlfornew的博客

10-27

1236

在将pytorch训练好的模型转成onnx模型时遇到最多的报错之一就是Expected object of device type cuda but got device type cpu这种数据计算位置的错误，这个错误的原因是在转成onnx模型时会将自动将原来代码中所有的python数值都转为tensor（不管是常量还是变量），在转为tensor之后肯定要牵扯到数据计算位置的问题，比如之前源代码是在GPU能正常运行但是有些数据是python数值，然后在转换onnx模型时这些python数值都转为了tens

Pytorch高版本(1.0)跑程序出现的几个错误

yo跟着新宇走

04-09

1万+

Pytorch高版本1.0跑程序出现的几个错误错误 output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]错误 NotADirectoryError: [WinError 267] 目录名称无效。错误 Expected object of scalar type Long but got scalar ...

RuntimeError: Expected object of backend CUDA but got backend CPU for argument #4 'mat1'

RuntimeError: Expected object of device type cuda but got device type cpu for argument #2

最后，成功，我真的是太小白了

before：

now：