apex/amp 安装问题解决

在gpu上的安装方式

官方地址：https://github.com/NVIDIA/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" ./
或者
pip install -v --disable-pip-version-check --no-cache-dir ./

一般第3行命令安装不起来。用第4行的。

今天在1080ti环境上使用时，amp报错说缺少amp_c。
ModuleNotFoundError: No module named ‘amp_C‘
网上看结论说
把报错的那个import注释掉即可。

具体原理可以探究。