windows下安装nvidia apex教程！实测通过！！！

最新推荐文章于 2024-07-10 18:08:33 发布

田伯光光

最新推荐文章于 2024-07-10 18:08:33 发布

阅读量2.3k

点赞数 2

本文链接：https://blog.csdn.net/hktxt/article/details/107543435

版权

本教程详细介绍了在Windows系统中安装NVIDIA Apex的步骤，包括安装PyTorch 1.5，修改torch和pybind11的源文件，安装VS 2019 Community，克隆并修改Apex项目，以及设置环境变量和使用pip安装。通过这些步骤，实测可以成功安装Apex库。

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

1. 安装pytorch 1.5

2. 修改：（根据自己的实际路径） /anaconda3/lib/site-packages/torch/include/torch/csrc/jit/api/module.h (474, 487, 501, 514) from static constexpr to static const

3. 修改：（根据自己的实际路径）/anaconda3/lib/site-packages/torch/include/pybind11/cast.h (1449) from explicit operator type&() { return *(this->value); } to explicit operator type&() { return *((type*)(this->value)); }

4. 安装VS 2019 community

5. 克隆 https://github.com/kezewang/apex 到本地

6. 修改 setup.py (104行) to version_dependent_macros = version_ge_1_1 + version_ge_1_3 + version_ge_1_5 + ['-D_ENABLE_EXTENDED_ALIGNED_STORAGE']

7. 命令行下执行一下：

set CMAKE_GENERATOR=Visual Studio 16 2019
set CMAKE_GENERATOR_TOOLSET_VERSION=14.26
set DISTUTILS_USE_SDK=1
for /f "