正常的安装流程,可参考(Pytorch 安装 APEX 疑难杂症解决方案)
$ git clone https://github.com/NVIDIA/apex
$ cd apex
$ pip install -v --no-cache-dir --global-option="--pyprof" --global-option="--cpp_ext" --global-option="--cuda_ext" ./
如果在执行第三行时出错 Cuda extensions are being compiled with a version of Cuda that does not...
,可以尝试使用python setup.py install
代替第三行的安装命令。安装完成如下。具体更多的问题,可查看如下链接。https://github.com/NVIDIA/apex/pull/323