apex github
- 下载apex的官方代码（git clone 太慢的话，直接手动去官网下载压缩包）
git clone 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" ./
可能出现的问题：
- RTX3090的算力太高，安装不成功，需要手动降低算力，输入下边的命令
export TORCH_CUDA_ARCH_LIST="7.5"
