安装apex步骤如下

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

踩坑

1. git clone 下载失败

https://github.com/NVIDIA/apex

打开以上的地址，将所有文件下载下来并解压

解压完成cd到setup.py所在目录下，然后运行如下命令：

python setup.py install --cpp_ext --cuda_ext

2. 安装cuda扩展失败

自己安装的cuda版本与安装pytorch所选的cuda版本需要一致，若因为这个原因安装失败，需要统一后再进行安装

3. 安装到使用的虚拟环境下

我使用的是conda管理的虚拟环境，在想要将apex安装到指定的虚拟环境中的情况下，使用**conda activate XXX（自己的虚拟环境名称）**再按照上述步骤安装即可。