首先最重要的是版本对应问题：
我的环境
Linux-Ubuntu 16.04.7 LTS
python=3.7
pytorch=1.4
cuda=10.1
上面四者的版本要对应，这个可以在网上查。

这是官方的安装步骤：

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" ./

如果还是不行，那么就把安装语句换成下面的：

python setup.py install --cuda_ext --cpp_ext

这里可能会报错：
FileNotFoundError: [Errno 2] No such file or directory: ‘:/usr/local/cuda-10.1/bin/nvcc’: ':/usr/local/cuda-10.1/bin/nvcc’

这里问题出在“: /“,所以我们只需要指明
export CUDA_HOME={/usr/local/cuda-10.1}去掉那个多余的“:”即可，再运行就没问题了，会报一些warning，不用管。