APEX安装

git clone https://github.com/NVIDIA/apex.git
cd apex

export CUDA_HOME=/usr/local/cuda
pip3 install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./