如何安裝NVIDIA/apex?

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Yanwei Liu
Aug 2, 2021

使用官方提供的安裝指令在aiForge上會遇到PyTorch及CUDA版本無法對應的問題,致使安裝錯誤。以下提供能成功安裝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" ./
# 能成功安裝的指令
$ git clone https://github.com/NVIDIA/apex
$ cd apex
$ python setup.py install

--

--