首先尝试了pip install amp,能够完成安装。但是安装后依然报错
解决方法:
1.首先amp是集成到apex包中,删掉apex包
2.重新安装apex包即可正常调用
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./