解决RuntimeError: gather_out_cuda(): Expected dtype int64 for index

原创于 2023-07-08 10:19:14 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #深度学习 #人工智能

在使用PyTorch训练模型时，遇到了一个RuntimeError，问题出在gather操作上，因为targetunsqueeze(2)的dtype是int32而不是expected的int64。将target的dtype转换为int64后，报错消除，程序得以正常运行。

部署运行你感兴趣的模型镜像

在训练模型过程中有如下语句，结果出现报错

output = -input.gather(2, target.unsqueeze(2)).squeeze(2) * mask

RuntimeError: gather_out_cuda(): Expected dtype int64 for index

原因应该是和pytorch版本有关系，根据报错，找到原因为target.unsqueeze(2)的dtype是int32类型的，把它转为int64了，可以正常运行。

您可能感兴趣的与本文相关的镜像

PyTorch 2.6

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

lingyunyuny

关注关注

0
点赞
踩
0

收藏

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

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

动手学强化学习-模仿学习RuntimeError: gather(): Expected dtype int64 for index

oceancoco的博客

12-06

1128

当出现RuntimeError这个错误信息的时候大概率是因为索引类型不对，追根溯源找到应该的索引类型，再找到错误源进行更改即可。即actions的类型转换为int64 经过输出发现原来的actions为int32类型的。经过不断查资料，发现gather只能索引int64类型的数据，所以将原来的。：原网站上的代码有一点问题，报错一下信息。

最近接触PyTorch 相关项目代码遇到的问题

qq_33511315的博客

12-29

1万+

问题出在scatter的函数参数上：第2个（算上self是第3个）参数应该是tensor，且其包含的元素类型应该是Long型，而原来程序内是Int型。应改为。

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

1 条评论

优快云-Ada助手 2023.09.03
首先，恭喜你开始了博客创作！对于解决这个特定的错误，你提供了一个令人期待的标题。在博客中，我希望你能更详细地解释这个错误的背景和原因，以及如何解决它。同时，你可以分享一些实际案例或者使用该错误的上下文，这将使读者更容易理解和跟随你的解决方案。希望你能继续保持谦虚的态度，并继续分享你的学习和经验！期待阅读更多精彩的博客！推荐【每天值得看】：https://bbs.youkuaiyun.com/forums/csdnnews?typeId=21804&utm_source=csdn_ai_ada_blog_reply1

torch.gather()

abc1831939662的博客

03-24

3102

作用:取出特定维度索引的值。 torch.gather(input,dim,index,out=None,sparse_grad=False) 举个例子 import torch a=torch.tensor([ [1,2,3,4,5], [6,7,8,9,10], [11,12,13,14,15], ]) index=torch.tensor([ [0,2], [3,4], [1,4] ]) print(torch.gather(a,1,index)) 输出 tensor([[ 1, 3]

gather_out_cuda(): Expected dtype int64 for index

zhongruizhe的博客

10-31

3791

gather_out_cuda(): Expected dtype int64 for index 在 Dual-Level Collaborative Transformer for Image Captioning复现代码的时候发现bug。问题原因：由于pytorch版本不对因为我的显卡是gtx3090,所以cuda版本只能是11.0，安装的pytorch1.7 解决思路：初始代码： selected_beam = selected_idx / candidate_logprob.shape[-1

stanfordnlp使用报错“index_select(): Expected dtype int32 or int64 for index”解决方法

AdrenalinmyVein的博客

05-21

4008

stanfordnlp使用报错（float index相关）解决方法

RuntimeError: gather_out_cuda(): Expected dtype int64 for index怎么解决？

11-14

RuntimeError: gather_out_cuda(): Expected dtype int64 for index 这个错误通常发生在使用PyTorch进行张量操作时，当你尝试通过`gather`函数从张量中选择元素，但是提供的索引数据类型不是预期的int64（64位整数）...

RuntimeError: gather_out_cpu(): Expected dtype int64 for index

热门推荐

weixin_45928096的博客

01-08

1万+

RuntimeError: scatter(): Expected dtype int64 for index 1.报错原因： scatter要求数据是int64类型，而我在定义tensor时写的是torch.Tensor(x)，应该写成torch.LongTensor(x)，指定为int64类型。 2.解决方法找到原数据的定义方式，改！一般在dtype=np.int64；dtype=np.float32中（多数定义函数都有dtype属性）最好int和float的位数要一致

torch.gather()的理解

qq_3311753064的博客

03-20

4143

在看蘑菇书的时候出现了这样的代码： '''计算当前(s_t,a)对应的Q(s_t, a)''' '''torch.gather:对于a=torch.Tensor([[1,2],[3,4]]),那么a.gather(1,torch.Tensor([[0],[1]]))=torch.Tensor([[1],[3]])''' q_values = self.policy_net(state_batch).gather(dim=1, index=action_batch) # 等

pytorch中index_select()

qq_37972486的博客

05-15

397

** pytorch中index_select()的用法 ** a = torch.linspace(1, 12, steps=12).view(3, 4) print(a) b = torch.index_select(a, 0, torch.tensor([0, 2])) print(b) print(a.index_select(0, torch.tensor([0, 2]))) c = torch.index_select(a, 1, torch.tensor([1, 3])) print(c)

torch中常见的错误信息

weixin_61745097的博客

11-12

5094

torch中运行过程中的常见的错误信息的分析与解决方案，希望可以帮到大家。

pytorch使用笔记

weixin_47895059的博客

07-25

901

1.torch.max() torch.max(input,0/1) 其中input是一个tensor，返回两个tensor，如果第二个参数是零，返回每行最大值和所在列数，反之对应 a = torch.tensor([[1,5,62,54], [2,6,2,6], [2,65,2,6]]) x=torch.max(a,1) y=torch.max(a,0) print(x) print(y) ...

pytorch gather_Pytorch之Tensor超详细理解（一）

weixin_39628063的博客

11-23

317

Tensor，又名张量，几乎所有的深度学习框架都有这种数据结构。下面对工程实现的方法进行详细介绍，若能精通Tensor，以后想建立自己的深度学习模型时，速度必将得到巨大的提升，同时也能加强对模型细节的理解。1. 分类从接口的角度来讲，对tensor的操作可分为两类：(1)torch.function，如torch.save(保存模型)等。(2)tensor.function，如tenso...

关于使用stanfordcorenlp一直运行不报错的解决方法

蜗牛之窝

06-26

3068

最近在使用stanfordcorenlp时，遇到了在运行代码后不报错但同时也没有任何结果输出的问题，等了很久也没有出结果。网上也没找到类似的问题反馈，还好在博客园一篇博文记录了这个问题，自己也实践了一下，确实如此，在此记录一下，希望遇到这个问题的同学能够少走些弯路~ 根本原因在于stanfordcorenlp是基于java开发的，而且需要的内存是4g，如果你的JAVA环境是32位的，而32位的j...

pytorch系列（一）常见错误

qq_30049011的博客

01-26

1万+

pytorch使用（一）常见错误 1、RuntimeError: scatter_(): Expected dtype int64 for index. RuntimeError: scatter_(): Expected dtype int64 for index. scatter要求数据是int64类型，而我在定义tensor时写的是torch.Tensor(x)，应该写成torch.LongTensor(x)，指定为int64类型。其他数据类型的定义方法可参考： https://pytorch.or

[pytorch]亲测解决RuntimeError: CUDA out of memory

万万VV

03-17

1万+

问题当我在测试训练好的基于Pytorch框架的半监督视频目标分割模型时，我已经加上了Model.eval( )，用于测试，但是运行过程中报错：RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 3.95 GiB total capacity; 3.39 GiB already allocated; 9.88...