Apex

Apex 是 Nvidia 提供的一个基于 PyTorch 的混合精度训练加速器，可以加速 PyTorch 的训练效率。

安装

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