记录一个bug解决
import collections.abc as container_abcs 替换 from torch._six import container_abcs
Reference:
[1] https://github.com/NVIDIA/apex/issues/1048
PS:实测APEX可以在CUDA10.2 + pytorch1.9.0的环境下使用

记录一个bug解决
import collections.abc as container_abcs 替换 from torch._six import container_abcs
Reference:
[1] https://github.com/NVIDIA/apex/issues/1048
PS:实测APEX可以在CUDA10.2 + pytorch1.9.0的环境下使用