git clone https://github.com/NVIDIA/apex
cd apex
服务器中有C++编译环境:
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
有python环境(推荐)
pip install -v --disable-pip-version-check --no-cache-dir ./