linux安装apex使用混合精度加速

官方安装方法：

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

 失败了，import apex时提示没有这个包。

git clone https://github.com/NVIDIA/apex
cd apex
python3 setup.py install

安装apex成功了，但是运行时报错 No module named 'amp_C'。

最后直接使用下面这段代码，终于成功了。

conda install -c conda-forge nvidia-apex

用apex加速之后，速度将近是原来的3倍，这被折磨的1h还是值得的。