RuntimeError: ‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor

🤵 Author ：Horizon John

✨ 编程技巧篇：各种操作小结

🎇 机器视觉篇：会变魔术 OpenCV

💥 深度学习篇：简单入门 PyTorch

🏆 神经网络篇：经典网络模型

💻 算法篇：再忙也别忘了 LeetCode

错误提示

使用 PyTorch 训练，CPU 可以正常使用，使用 GPU 时出现以下错误：

RuntimeError: 'lengths' argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor

错误原因

PyTorch 版本的问题，版本大于1.5 会报以上错误

解决方案

以我的为例：

解决前 ：

output = classifier(inputs, seq_lengths)

解决后 ：

output = classifier(inputs, seq_lengths.to('cpu'))

总结

找 到 报 错 的 行 数

将 length 改成 length.to('cpu') 即可

🈺 喜欢的 留个 关注 、 加 点赞 哦 ~