为pytorch安装nvidia的apex加速

发表于 2019-10-10 更新于 2020-05-07 阅读次数：
本文字数： 215 阅读时长 ≈ 1 分钟

三个指令，不需要配置环境变量

1
2
3

$ git clone https://github.com/NVIDIA/apex
$ cd apex
$ pip3 install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./