ubuntu安装apex

apex的两种使用方式

1.NVIDIA apex

apex库前一定要确保pytorch和cuda版本适配

nvcc -V

python
import torch
print(torch.version.cuda)

不能通过pip install apex直接安装，安装步骤如下：

git clone https://github.com/NVIDIA/apex
cd apex
pip install najia #安装najia等依赖库
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

安装成功后界面如下：

2.pytorch>1.6原生支持apex（没有尝试使用）