一文详解Apex的安装和使用教程（一款基于 PyTorch 的混合精度训练加速神器）

前言： 本篇博文主要介绍Apex的安装和使用教程（一款基于 PyTorch 的混合精度训练加速神器）

win 10系统下安装NVIDIA apex

1. 首先在github下载源码https://github.com/NVIDIA/apex 到本地文件夹
2. 打开cmd命令窗口，切换到apex所在的文件夹
3. 使用命令：python setup.py install 即可完成安装

注意事项： 可能会出现的问题：
setuptools有ModuleNotFoundError→更新setuptools

pip install --upgrade setuptools

linux系统下安装NVIDIA apex

安装操作如下所示：

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