Colab Notebook：无法从“torch._six”导入名称“container_abcs”

Question

我正在尝试运行此处找到的 deit colab 笔记本：

https://colab.research.google.com/github/facebookresearch/deit/blob/colab/notebooks/deit_inference.ipynb

但是我在第二个单元格中遇到了问题，特别是

import timm

行，它返回以下内容：

导入错误：无法从“torch._six”导入名称“container_abcs”

Answer 1

• 与此错误相关的问题此处：

尝试 timm 库的特定

版本

：

!pip install timm==0.3.2

Answer 2

就我而言，它适用于

pip install timm==0.4.12

Answer 3

当我安装 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

Answer 4

我在使用 PyTorch 1.6 开发的代码库开发 PyTorch 1.12 版本时遇到了这个问题。

你应该更换

from torch._six import container_abcs 

与

import collections.abc as container_abcs 

按照 GitHub 上的此问题。