官方安装方法:
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还是值得的。