【pytorch错误】：Pytorch RuntimeError: “host_softmax” not implemented for 'torch.cuda.LongTensor'

最新推荐文章于 2023-03-13 20:52:12 发布

原创最新推荐文章于 2023-03-13 20:52:12 发布 · 1.5w 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习

深度学习专栏收录该内容

30 篇文章

订阅专栏

本文详细解析了PyTorch运行时错误“host_softmax”未实现的问题，指出训练标签train_y应为int64类型，并提供了修改数据类型的具体代码示例。

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

问题

Pytorch RuntimeError: “host_softmax” not implemented for ‘torch.cuda.LongTensor’

报错的位置在这个地方

    loss=criterion(out,train_y)  # train_y 应该是int64

原因

参考：https://stackoverflow.com/questions/51818225/pytorch-runtimeerror-host-softmax-not-implemented-for-torch-cuda-longtensor。
大致就是说，train_y应该是int64类型.

解决

在将train_y变为一个tensor的时候，设置数据类型为int64。

    train_x=torch.tensor(train_x,dtype=torch.float32).cuda()
    train_y=torch.tensor(train_y,dtype=torch.int64).cuda()
    train_x = Variable(train_x)
    train_y = Variable(train_y)