ImportError: cannot import name ‘container_abcs‘ from ‘torch._si
【摘要】 问题描述在使用apex的过程中,出现了如下问题: File "/home/shuyuan/yolov5-master/./fast_reid/fastreid/engine/train_loop.py", line 14, in <module> from apex import amp File "/home/shuyuan/anaconda3/envs/shuyuan/lib...
问题描述
在使用apex的过程中,出现了如下问题:
File "/home/shuyuan/yolov5-master/./fast_reid/fastreid/engine/train_loop.py", line 14, in <module>
from apex import amp
File "/home/shuyuan/anaconda3/envs/shuyuan/lib/python3.8/site-packages/apex/__init__.py", line 7, in <module>
from . import amp
File "/home/shuyuan/anaconda3/envs/shuyuan/lib/python3.8/site-packages/apex/amp/__init__.py", line 1, in <module>
from .amp import init, half_function, float_function, promote_function,\
File "/home/shuyuan/anaconda3/envs/shuyuan/lib/python3.8/site-packages/apex/amp/amp.py", line 1, in <module>
from . import compat, rnn_compat, utils, wrap
File "/home/shuyuan/anaconda3/envs/shuyuan/lib/python3.8/site-packages/apex/amp/rnn_compat.py", line 1, in <module>
from . import utils, wrap
File "/home/shuyuan/anaconda3/envs/shuyuan/lib/python3.8/site-packages/apex/amp/wrap.py", line 3, in <module>
from ._amp_state import _amp_state
File "/home/shuyuan/anaconda3/envs/shuyuan/lib/python3.8/site-packages/apex/amp/_amp_state.py", line 14, in <module>
from torch._six import container_abcs
ImportError: cannot import name 'container_abcs' from 'torch._six' (/home/shuyuan/anaconda3/envs/shuyuan/lib/python3.8/site-packages/torch/_six.py)
解决方法
找到问题所在的文件,将这部分的代码用下面的代码替换
if TORCH_MAJOR == 1 and TORCH_MINOR < 8:
from torch._six import container_abcs
else:
import collections.abc as container_abcs
【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)