ImportError: No module named apex

也可以直接下载文件夹 import https://github.com/nvidia/apex#__NO_LINK_PROXY__

git clone https://www.github.com/nvidia/apex
cd apex
python setup.py install [--cuda_ext] [--cpp_ext]