Pytorch错误：AssertionError: If capturable=False, state_steps should not be CUDA tensors

马鹤宁

于 2023-10-09 14:42:09 发布

阅读量1.2k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：机器学习和深度学习之旅文章标签： pytorch 人工智能 python

84 篇文章

订阅专栏

文章讲述了在使用PyTorch1.12.0和CUDA11.3进行训练时遇到的AssertionError，由于state_steps不应为CUDATensors导致中断。通过降级到PyTorch1.11版本解决问题，同时保持了较高的训练效率。

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

基础配置的版本信息如下所示：

现象描述：训练中断，从中断处再起训练时，爆出AssertionError: If capturable=False, state_steps should not be CUDA tensors的错误。

出错原因

解决方案：在stackOverflow中找到的答案。
在这里插入图片描述
不想降低训练速度，我将pytorch卸载，安装1.11版本的pytorch，解决了问题。