安装nvidia apex, g++ cuda torch版本必须都匹配才行

pip install apex安装的不是nvidia的apex

需要到https://github.com/NVIDIA/apex下载，如果github打不开，可以用gettee拉取到国内下载

解压apex，进入apex文件夹，输入

pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

运气好就安装好了，运气不好就很多问题

1.

RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries. Pytorch binaries were compiled with Cuda 10.1.
这个是cuda版本和pytorch的cuda版本不一致导致的，可能装了2个cuda，需要更改默认cuda，或者直接把setup.py189行的check_cuda_torch_binary_vs_bare_metal(CUDA_HOME)注释掉，但是可能有未知问题，我是直接重装了cuda。

进https://developer.nvidia.cn/cuda-toolkit-archive选择对应的版本，linux安装很快，几分钟就好了。

2. nvcc找不到，这个默认是在/usr/local/cuda/bin/nvcc,可以重装cuda或者设置一下环境变量就可以了。

3.gcc版本不对，我试过很多版本，默认的4不行，升级5.2也不行，升级9.3也不行，后来换到7就可以了。

不同版本的linux升级gcc方法不一样，就不细说了。

装好了需要export CXX="/usr/bin/g++"，指定新安装的g++,可以改成新的位置。