apex混合精度训练 安装步骤

apex github

1. 下载apex的官方代码（git clone 太慢的话，直接手动去官网下载压缩包）

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

2. 进入文件

cd apex

3. 安装

pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

可能出现的问题：

1. RTX3090的算力太高，安装不成功，需要手动降低算力，输入下边的命令

export TORCH_CUDA_ARCH_LIST="7.5"