前言: 本篇博文主要介绍Apex的安装和使用教程(一款基于 PyTorch 的混合精度训练加速神器)
win 10系统下安装NVIDIA apex
首先在github下载源码https://github.com/NVIDIA/apex 到本地文件夹
打开cmd命令窗口,切换到apex所在的文件夹
使用命令: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" -