pytorch Colab笔记本:无法从“torch._six”导入名称“container_abcs”

70gysomp  于 2023-05-07  发布在  其他
关注(0)|答案(4)|浏览(295)

我正在尝试运行在这里找到的deit colab notebook:
https://colab.research.google.com/github/facebookresearch/deit/blob/colab/notebooks/deit_inference.ipynb
但是我在第二个单元格中遇到了一个问题,特别是import timm行,它返回以下内容:
ImportError:无法从“torch._six”导入名称“container_abcs”

piv4azn7

piv4azn71#

  • 与此错误here相关的问题:

尝试timm库的特定 * 版本 *:

!pip install timm==0.3.2
mzillmmw

mzillmmw2#

当我安装torch==1.9.0和torch-geometric时,旧代码有错误。
以下是我解决方案:

TORCH_MAJOR = int(torch.__version__.split('.')[0])
   TORCH_MINOR = int(torch.__version__.split('.')[1])
   if TORCH_MAJOR == 0:
      import collections.abc as container_abcs
   else:
      from torch._six 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,int_classes
    else:
        import collections.abc as container_abcs
        int_classes = int
ckx4rj1h

ckx4rj1h3#

在我的情况下,它与

pip install timm==0.4.12
mzsu5hc0

mzsu5hc04#

我在使用基于PyTorch 1.6开发的代码库开发PyTorch 1.12版本时遇到了这个问题。
你应该换掉

from torch._six import container_abcs

import container.abc as container_abcs

根据this issue on GitHub

相关问题