Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU

最新推荐文章于 2024-03-20 10:53:46 发布
最新推荐文章于 2024-03-20 10:53:46 发布
阅读量3.1k 收藏 1
点赞数 3
分类专栏： pytorch知识宝典 文章标签： pytorch
原文链接：https://blog.csdn.net/weixin_43760844/article/details/116047427
版权
在使用PyTorch进行模型测试时，遇到RuntimeError，提示输出张量在CPU上，而模型参数张量也在CPU上，但期望它们都在GPU上。问题源于先将模型移动到GPU，然后加载在CPU上的预训练模型。解决方法是调整加载模型的顺序，确保模型移动到GPU后再加载权重。
摘要由CSDN通过智能技术生成

1、问题
模型训练完后进行测试，报错

RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm)

2、原因
将模型送入GPU之后才加载之前训练好的模型，导致模型加载到了GPU，然而你的网络权重还在CPU中，此时会报错如下，报错部分代码：

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
net.to(device)
net.load_state_dict(torch.load('./results/bestmodel30.pth'),False)



3、解决
换下位置即可


net.load_state_dict(torch.load('./results/bestmodel30.pth'),False)
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
net.to(device)



原文链接：https://blog.csdn.net/weixin_43760844/article/details/116047427

AI算法网奇
关注
专栏目录
订阅专栏
Tensor for argument #2 ‘mat1‘ is on CPU, but expected it to be on GPU (while checking arguments for
weixin_41938996的博客
07-07 5692
Tensor for argument #2 ‘mat1’ is on CPU, but expected it to be on GPU (while checking arguments for addmm) 需要将模型和输入数据都移动到 device model=NonLinearRegression().to(device)#模型 for batch_idx,(data,target) in enumerate(train_loader): data,target=data.to(device),
RuntimeError Expected tensor for argument #1 'input' to have the same device as tensor for argument
Jemary_的博客
10-23 7641
RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 1 does not equal 0 (while checking arguments for cudnn_convolution) 记录一下bu...
RuntimeError: Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU, but expected them
qq_40471091的博客
03-13 3944
RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm 产生这类错误的原因,模型和数据未在相同设备上 解决方法: 将其放在同一设备上,模型和数据都放在GPU上 net .to(“cuda”) data.to(“cuda”) 最后如果还出现相同错误,请检查网络模
解决RuntimeError: Tensor for ‘out’ is on CPU, Tensor for argument #1 ‘self’ is on CPU, but expected th
yimenren的博客
12-14 4277
遇到上述错误可能是由于没有将所有变量搬到GPU上去，你可以尝试这样做，并且一定要注意顺序 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') g, features, labels, mask = load_cora_data() g = g.to(device) #注意这里顺序不能乱，g进行转换后再赋给GAT net = GAT(g, in_dim=features.size()[1], hidden_dim=8
解决：RuntimeError: Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU, but expected t
地中海の养成记
04-23 1万+
这里写目录标题一级目录二级目录三级目录 一级目录 RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm) 二级目录 三级目录
【PyTorch问题】Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU，but expected them...略
冰糖的小屋
04-04 6456
问题描述：RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm) 问题详情：model传到了GPU，但是数据没有传到GPU 解决方法：将所需数据传至GPU，比如 batch_data= batch_data.to("cuda") ...
RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument
guyejiyou64的博客
10-11 9185
RuntimeError: Expected tensor for argument #1 ‘input’ to have the same device as tensor for argument #2 ‘weight’; but device 0 does not equal 1 (while checking arguments for cudnn_convolution) pytorch...
pytorch 使用gooleNet中的Inception模块训练损失不下降，使用显卡提示
weixin_42702793的博客
01-13 664
Tensor for ‘out’ is on CPU, Tensor for argument #1 ‘self’ is on CPU, but expected them to be on GPU (while checking arguments for addmm) 检查发现是Net类中出现了问题。 错误模块代码 class Net(torch.nn.Module): def __init__(self): super(Net,self).__init__()
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: Tensor for argument #2 ‘mat1‘ is on CPU, but expected it to be on GPU (
子根的博客
07-20 3647
RuntimeError: Tensor for argument #2 'mat1' is on CPU, but expected it to be on GPU TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
成功解决RuntimeError: Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU, but expected
最新发布
创作高质量博文，分享知识，共同进步！
03-20 1463
🚀 深度学习遭遇GPU张量位置不匹配问题？别慌！本文将带你一探究竟，解决RuntimeError的烦恼！ 🔍 深入剖析问题根源，并教你轻松将Tensor从CPU搬到GPU，确保所有操作都在同一设备上执行。 💡 深度学习模型训练也能飞速提升！掌握在GPU上运行模型的技巧，让计算加速不再是梦。 🔍 避开常见错误，确保张量管理得心应手。 🚀 总结经验，举一反三，让PyTorch使用更得心应手！ #PyTorch #GPU #张量位置不匹配 #深度学习模型训练 #Tensor操作
pytorch踩坑记录：
u011489220的博客
03-06 2633
1.出现不能分配内存错误： fork OSError: [Errno 12] Cannot allocate memory 解决办法： 重启电脑； 或者换用spaw启动方式 添加代码： mp.set_start_method('spawn')
torch.nn.MaxPool2d()学习笔记
敲代码的小风
12-21 4269
参考链接: torch.nn.MaxPool2d(kernel_size, stride=None, padding=0, dilation=1, return_indices=False, ceil_mode=False) 参考链接: Convolution arithmetic 参考链接: 神经网络与深度学习 参考链接: 二维转置卷积和空洞卷积示例
PyTorch中的torch.nn.Linear详解
热门推荐
qq_44289607的博客
01-30 1万+
torch.nn是包含了构筑神经网络结构基本元素的包，在这个包中，可以找到任意的神经网络层。这些神经网络层都是nn.Module这个大类的子类。torch.nn.Linear就是神经网络中的线性层，可以实现形如y=Xweight^T+b的加和功能。 其用法与形参说明如下： 代码很简单，但有许多细节需要声明： 1）nn.Linear是一个类，使用时进行类的实例化 2）实例化的时候，nn.Linear需要输入两个参数，in_features为上一层神经元的个数，out_features为这一层的神经
pytorch多gpu的问题（RuntimeError: Expected tensor for argument #1 ‘input‘ to have the same device as ...）
weixin_43508499的博客
03-02 1万+
目录1. pytorch正常使用多GPU的方式2. 报错：RuntimeError: Expected tensor for argument #1 'input' to have the same device as ...2.1 问题描述2.2 问题分析3. 总结 1. pytorch正常使用多GPU的方式 # 导入需要的库 import torch import torch.nn as nn...
RuntimeError: Tensor for argument #2 ‘weight’ is on CPU, but expected it to be on GPU (while checkin
weixin_45104871的博客
09-03 1499
model = Network() model.to(device) 链接
错误记录帖(持续更新)
喵喵喵
10-21 1714
错误记录帖
ToTensor.__init__() takes 1 positional argument but 2 were given
06-10
这个错误通常是因为在调用`ToTensor()`函数时，传递了不必要的参数导致的。`ToTensor()`函数只需要在初始化时传递一个`self`参数即可，不需要再传递其他参数。请检查你的代码，确保在调用`ToTensor()`函数时，只传递了一个`self`参数。如果还有其他问题，请提供更多的代码和上下文信息，以便更好地帮助你解决问题。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值