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