使用MMSelfSup时，报错If capturable=False, state_steps should not be CUDA tensors.

最新推荐文章于 2024-08-10 08:25:15 发布

愿与愁

最新推荐文章于 2024-08-10 08:25:15 发布

阅读量253

点赞数 1

文章标签： linux python 人工智能

本文链接：https://blog.csdn.net/m0_57143158/article/details/133063852

版权

文章讲述了用户在使用MMSelfSup进行模型训练时遇到的错误，原因是Pytorch1.12.0版本的bug。为解决问题，作者降级到Pytorch1.9.0并重新安装，同时提到MMCV版本需与Pytorch和CUDA版本匹配。

摘要由CSDN通过智能技术生成

1、使用MMSelfSup时，模型正常训练都没问题。但是想按照之前训练的结果，继续接着训练时报错

If capturable=False, state_steps should not be CUDA tensors.

2、查了一下是Pytorch1.12.0版本的bug，无奈只能换个其他版本的Pytorch。我使用以下语句重新安装了Pytorch 1.9.0，然后就不报错了。

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

3、我是把MMSelfSup虚拟环境删除了，重新安装了一遍。我害怕如果仅在虚拟环境中把之前的Pytorch卸载了，使用以上语句安装了新的Pytorch后，又与MMCV版本不对应，报其他错误（MMCV必须与Pytorch与cuda的版本对应。MIM安装MMCV时会根据Pytorch与CUDA的版本，自动选择合适的MMCV）。