PyTorch中apex的安装方式

apex是NVIDIA开发的基于PyTorch的混合精度训练加速神器，能够增加运算速度，并且减少显存的占用。

• Github地址：https://github.com/NVIDIA/apex
• 官方教程：https://nvidia.github.io/apex/

安装方式

需要注意的是apex的安装不能通过 pip install 的方式安装（pip install apex安装的是两个完全不同的库），需要基于Github源码安装。官方提供的安装方式如下：

git clone https://github.com/NVIDIA/apex
cd apex
# if pip >= 23.1 
pip install -v --disable-pip-version-check <