cannot import name ‘container_abcs‘ from ‘torch._six‘错误的解决方法

最新推荐文章于 2024-07-25 19:08:19 发布

Agno ni

最新推荐文章于 2024-07-25 19:08:19 发布

阅读量64

点赞数 3

文章标签：深度学习人工智能 pytorch

本文链接：https://blog.csdn.net/weixin_53206200/article/details/140636912

版权

出错原因：当pytorch的版本到2.0以后就不在集成container_abcs所以：

from torch import container_abcs

这句话就是错的，可以改成

# from torch import container_abcs
TORCH_MAJOR = int(torch.__version__.split('.')[0])
TORCH_MINOR = int(torch.__version__.split('.')[1])
if TORCH_MAJOR == 1 and TORCH_MINOR < 8:
    from torch._six import container_abcs
else:
    import collections.abc as container_abcs