基础配置的版本信息如下所示:
| 版本号 |
---|
pytorch | 1.12.0 |
cuda | 11.3 |
python | 3.10 |
现象描述:训练中断,从中断处再起训练时,爆出AssertionError: If capturable=False, state_steps should not be CUDA tensors
的错误。
出错原因
解决方案:在stackOverflow中找到的答案。
不想降低训练速度,我将pytorch卸载,安装1.11版本的pytorch,解决了问题。
参考
- AssertionError: If capturable=False, state_steps should not be CUDA tensors