🤵 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')
即可
![](https://i-blog.csdnimg.cn/blog_migrate/e4bfbafa14994d7da6e2f43f03db4ebd.gif#pic_center)
🈺 喜欢的 留个 关注
、 加 点赞
哦 ~