Python - 安装Apex问题

下载Apex工程：

git clone https://github.com/NVIDIA/apex && cd apex

安装命令

pip3 install --user -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

参考 PyTorch 安装 APEX 疑难杂症解决方案:

https://zhuanlan.zhihu.com/p/80386137

问题：ModuleNotFoundError: No module named ‘torch’

安装相应版本的PyTorch即可。

问题：如果在执行第三行时出错 “Cuda extensions are being compiled with a version of Cuda that does not…”, 尝试一下解决方案。

vim setup.py

更改第 52 行代码，注释掉RuntimeError即可。

PyTorch1.4.0 需要 GCC5.0+，PyTorch1.3.0 需要 GCC4.9+