nvidia-apex cuda 版本安装

当前apex 版本：0.1
需要环境:

• gcc-6.5/g++ 6.5
• 系统cuda版本要与pytorch 的cuda 版本完全一致:
  如果版本pytorch 对应的是cu101,系统里cuda版本是cu100,运行pytorch 没有问题但是安装apex就会报错,更改系统，cuda版本，更改后需要重新安装pytorch

安装apex

git clone https://github.com/NVIDIA/apex
cd ./apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext"

refer from :

• nvidia apex official github
• https://blog.csdn.net/Diana_Z/article/details/89496114