1 官方github （https://github.com/NVIDIA/apex）

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

报错是因为基础环境有误，参考
https://blog.csdn.net/justforacm/article/details/115428583

2 来自知乎（linux安装nvidia/apex - 知乎）

git clone https://github.com/NVIDIA/apex
cd apex
python3 setup.py install
python setup.py install --cuda_ext --cpp_ext

3. 如果还不行

export TORCH_CUDA_ARCH_LIST="compute capability" 
conda install -c nvidia cuda-nvcc

或

cd apex-22.04-dev
pip install -r requirements.txt
pip install -r requirements_dev.txt
pip install -v --disable-pip-version-check --no-cache-dir ./