Pytorch出现RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)

最新推荐文章于 2024-06-17 16:39:34 发布
最新推荐文章于 2024-06-17 16:39:34 发布
阅读量1.2k 收藏 3
点赞数 3
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_39123145/article/details/109224591
版权

一般来说遇到这个问题可能是没有执行：

model.cuda()

或者：

model.to('cuda')

 

但是我这次遇到的并不是因为以上原因，而是因为:

self.head_convs = [ ]

这里应该用 nn.ModuleList 而不是普通的 python list。使用后者会导致 self.head_convs 中存放的nn.Module 无法进入GPU。因此，应该改成如下：

self.head_convs = nn.ModuleList()

Thinker_and_FKer
关注
  • 3
    点赞
  • 踩
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
pytorch报错系列（1）RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)
weixin_44776845的博客
03-15 1933
在训练期间，运行如下代码 import torch import torch.nn as nn from lenet import LeNet, MyNet from torchsummary import summary net = LeNet(classes=2) net.initialize_weights() summary(net, input_size=(3, 32, 32)) 报...
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
01-20
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够 简单总结一下解决方法： 将batch_size改小。 取torch变量标量值时使用item()属性。 可以在测试阶段添加如下代码：...
pytorch RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)
荪荪的博客
09-26 649
报错： RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same 分析原因： cpu和cdua 使用的不一致，比如说你的模型和输入，一个在cpu上，一个在cuda上，导致这样的错误。 解决方法： net = Model(cfg).to(device) net.load_state_dict(torch.load(w2)) 中的device，看输入的img是在
莫名遇到RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED的解决办法
最新发布
huashuicaigou的博客
06-17 186
我使用3090Ti，torch版本对应的CUDA是10.2，此时运行报错。安装对应11.2的torch版本问题解决。因为3090Ti不支持CUDA 10.x。这种问题一般都是不是代码错误，如果可以排查代码本身问题之后，应该注意CUDA torch GPU硬件型号的多种问题。
python 报错：Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
weixin_40893448的博客
04-26 2501
需要在网络模型model = MSTransception(num_classes=9)后面添加设备即可，即；model = MSTransception(num_classes=9).cuda()则需要在网络模型model = MSTransception(num_classes=9)后面添加设备，即；为了检测网络是能够正常执行时，如果运行整个代码则会导致不知道问题出在哪且不利于排查错误，因此，我们只需将所提模型单独进行进行检测以确保所提模型是正确的。
Error: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
qq_37683597的博客
12-11 697
@[TOC](RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same) pytorch使用中出现以上问题解决办法 以上错误是因为部分网络部分模块用列表list放置，没有被nn.ModuleList()处理；导致pytorch无法识别，并放到cuda上； 利用nn.ModuleList()可解决： class OneModule(nn.Module): de
报错RuntimeError: torch.cuda.FloatTensor is not enabled.的可能原因及解决办法
zeeq的博客
10-28 4386
  使用pytorch时候报错RuntimeError: torch.cuda.FloatTensor is not enabled.，具体信息如下所示： 可能原因1   出现这个错误，有可能是因为你的电脑没有英韦达显卡，不支持GPU加速，而你在代码中使用了gpu。解决办法就是把gpu版本的代码改成cpu版本的，也就是把代码里所有出现.cuda()的地方都修改为.cpu()即可。具体可以参考这里。 可能原因2   还一个原因，刚好和上面的反过来了。当然，不是因为电脑没有cpu，而是你下载安装的pytorc
linux或windows环境下pytorch的安装与检查验证(解决runtimeerror问题)
01-10
但是其实这样比较容易断或者出现runtimeout的error。解决办法：使用pip先进行源的查找，在开始下载后停掉，然后找到屏幕上打印出来的链接，手动下载whl 然后手动安装。 方法二：选择相应版本的whl进行离线下载 找到...
RuntimeError: DataLoader worker (pid(s) 9528, 8320) exited unexpectedly
01-06
RuntimeError: DataLoader worker (pid(s) 9528, 8320) exited unexpectedly 去网上搜教程，说是在loader定义的时候多进程的参数指定的问题，我的loader定义如下 loader = Data.DataLoader( dataset=torch_dataset,...
pytorch dataloader 取batch_size时候出现bug的解决方式
12-20
1. `RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 342 and 281 in dimension 3` 2. `RuntimeError: invalid argument 0: Sizes of tensors must match except in ...
【PYTORCH】RuntimeError: torch.cuda.FloatTensor is not enabled.
m0_58810879的博客
12-29 1571
这个错误的原因是你在安装了cpu版本的pytorch上运行GPU版本的pytorch的代码。 解决方法： 将出错位置的".cuda()"换成".cpu()"
博客使用方法
justwaityou1314的博客
06-09 1565
这里写自定义目录标题欢迎使用Markdown编辑器新的改变功能快捷键合理的创建标题，有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能，丰富你的文章UML 图表FLowchart流程图导出与导入导出导入 欢迎使用Markdown编辑器 你好！ 这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Mar
成功解决TypeError: cannot assign ‘torch.cuda.FloatTensor‘ as parameter ‘weight‘ (torch.nn.Parameter or N
创作高质量博文，分享知识，共同进步！
04-24 4343
成功解决TypeError: cannot assign ‘torch.cuda.FloatTensor‘ as parameter ‘weight‘ (torch.nn.Parameter or N
torch.cuda.FloatTensor 与 torch.FloatTensor（torch.Tensor）--CPU和GPU上的数据类型
热门推荐
简单记录生活、学习
06-04 3万+
1.torch.cuda.FloatTensor 与torch.FloatTensor Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型，一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。 一般系统默认是torch.FloatTensor类型（即CPU上的数据类型）。例如data = torch.Tensor(2,3)是一个2*3的张量，类型为FloatTensor; data.cuda()就转换为GPU的张量类型，torch.cuda.FloatT.
[pytorch] TypeError cannot assign torch.FloatTensor as parameter weight
nc101100的博客
05-29 2万+
【pytorch】TypeError: cannot assign ‘torch.FloatTensor’ as parameter ‘weight’ (torch.nn.Parameter or None expected) 在尝试赋值线性层权重时候出现的错误 错误定位 def __init__(self, input_dim, atten_dim, attribute, classifier): super(MODEL,self).__init__() nc
Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
扇扇子的炼丹师
10-13 513
常规解决方案 从报错问题描述中可以找到错误原因 输入的数据类型为torch.cuda.FloatTensor，说明输入数据在GPU中 模型参数的数据类型为torch.FloatTensor，说明模型还在CPU 问题原因搞清楚了，模型没加载到CPU，在代码中加一行语句就可以了 model = model.cuda() model = model.to('cuda') model.cuda() model.to('cuda') 上面四行任选一 问题是加在哪里 一般在summary模型的时候提示错误， 所以可以
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the
P13643822101的博客
08-13 1万+
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the
runtimeerror: input type (torch.cuda.floattensor) and weight type (torch.flo
09-18
此错误是由于输入类型(torch.cuda.FloatTensor)和权重类型(torch.FloatTensor)不匹配所致。PyTorch中的张量类型包括CPU上的FloatTensor和GPU上的CUDA FloatTensor。在此情况下，输入张量被放置在GPU上，而权重张量却...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值