Windows环境下pip安装apex库

代码小白的成长

已于 2025-06-30 11:34:01 修改

阅读量177

点赞数 3

CC 4.0 BY-SA版权

文章标签： pip windows

于 2025-06-30 11:19:21 首次发布

本文链接：https://blog.csdn.net/weixin_43723625/article/details/149019111

在这里插入图片描述

未解为解决上面的问题，可执行下面的命令：

git clone https://www.github.com/nvidia/apex
cd apex
git checkout f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0
python setup.py install
pip install -v --no-cache-dir ./

在这里插入图片描述

在执行from apex import amp命令时，可能会出现下面的问题：

ImportError: cannot import name 'container_abcs' from 'torch._six'

解决：定位到出错的位置，将from torch._six import container_abcs注释，替换成import collections.abc as container_abcs。这是由于我安装的torch是1.8版本以上的，但是在1.8版本之后container_abcs就已经被移除了，所以导入方式不同会出现上面的错误。
在这里插入图片描述