NVIDIA apex安裝

如果是在docker中安裝，步驟如下：

（1）docker pull pytorch/pytorch:1.8.1-cuda10.2-cudnn7-devel。這里必須是-devel后綴！因為編譯時需要NVCC，只有-devel中有NVCC。cuda11.2有問題。

（2）按着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" ./