- 下载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"