Apex

Apex It is a PyTorch-based mixed precision training accelerator provided by Nvidia, which can accelerate the training efficiency of PyTorch.

Install

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" ./