Windows系统下安装apex，及错误汇总

博博有个大大大的Dream

已于 2022-11-11 23:05:12 修改

阅读量7.9k

点赞数 18

分类专栏：环境配置文章标签： windows python 开发语言

于 2022-11-11 22:10:41 首次发布

本文链接：https://blog.csdn.net/qq_17783559/article/details/127813381

版权

环境配置专栏收录该内容

50 篇文章 13 订阅

订阅专栏

本文档介绍了在Windows系统中安装NVIDIA apex库时可能遇到的问题及解决方案。尝试通过pip直接安装、从源代码编译安装等方法，但每个方法都可能遇到如CUDA版本不兼容、找不到文件、编译错误等问题。最终，通过特定版本的git检出并编译安装成功。

摘要由CSDN通过智能技术生成

官网地址：NVIDIA/apex: A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch (github.com)https://github.com/NVIDIA/apex
官方说支持Ubuntu系统，Windows只是实验性的，所以在windows上安装可能会遇到各种问题，本文不能涵盖所有问题。

方法1：直接安装

pip install apex

此指令会安装最新的apex 0.9以上版本，我遇到了如下错误：

WARNING: Building wheel for cryptacular failed: [Errno 2] No such file or directory: 'C:\\Users\\XXX\\AppData\\Local\\Temp\\pip-wheel-0dhfjrr3\\cryptacular-1.6.2-cp37-cp37m-win_amd64.whl'
Failed to build cryptacular
ERROR: Could not build wheels for cryptacular, which is required to install pyproject.toml-based projects

方法2：下载安装

git clone https://www.github.com/nvidia/apex
cd apex
python setup.py install

这样会安装apex 0.1版本，理想情况下就安装成功了，我遇到了错误汇总：

module 'torch.nn' has no attribute 'backends'  这个应该是CUDA版本和apex兼容问题

或者错误

AttributeError: module 'torch.distributed' has no attribute '_all_gather_base'

方法3：下载安装

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

或者
python setup.py install --cuda_ext --cpp_ext ./

还是遇到错误，

1 error detected in the compilation of "csrc/multi_tensor_scale_kernel.cu"

或者错误

ImportError was: ModuleNotFoundError("No module named 'amp_C'")

方法4：成功版本

git clone https://github.com/NVIDIA/apex.git
cd apex
git checkout f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0
python setup.py install --cuda_ext --cpp_ext