apex安装报错

错误1：安装时报错，import时找不到apex

大部分apex安装报错主要是因为环境不适配，例如cuda版本不适配torch，安装前注意查看需求的cuda版本:

# pytorch内的cuda版本
import torch
torch.version.cuda

具体安装参考：Nvidia Apex安装与简单使用_豆芽菜-CSDN博客

错误2：import时报apex ImportError: cannot import name 'container_abcs' from 'torch._six'

在_amp_state.py或_initialize.py中（取决于apex版本）使用

import collections.abc as container_abcs

而不是

from torch._six import container_abcs

例如在_amp_state.py中做如下修改：

#if TORCH_MAJOR == 0:
#    import collections.abc as container_abcs
#else:
#    from torch._six import container_abcs
import collections.abc as container_abcs