在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注释掉即可。
具体原理可以探究。