【备忘录】torch171+cu110配置apex

最近在复现项目时好几次用到apex进行浮点运算加速，之前都是胡乱找教程安装的，再安装时又得重新走一遍，今天记录一下留个备忘

首先，本文是以NVIDIA/apex项目下相关的issue为指导的，如果有朋友torch版本不是上述的torch171+cu110，可自行到下述链接搜索。

https://github.com/NVIDIA/apex/issues/1532

然后，再把主要的库版本列一下

python                    3.8.13 
torch                     1.7.1+cu110              pypi_0    pypi
torchaudio                0.7.2                    pypi_0    pypi
torchvision               0.8.2+cu110              pypi_0    pypi
tqdm                      4.64.1  

接下来就是安装步骤了，安装torch框架时Linux、Windows无差别。如果不清楚深度学习环境（CUDA、Pytorch）如何配置，可以参考我之前的博文：Pytorch环境配置

1 torch171 installation

pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

2 clone repo from github

注意，这里要想版本适配，必须下载22.04-dev。版本选择示意图如下：

必须先切换到指定版本下再git，否则默认下载最新版本

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

3 apex installation

安装apex过程中可选择纯python版本和C++依赖版本，C++配置好的可安装C++版，没有C++环境的安装纯python效果（运算速度）相差不大，安装过程中直接进入下载好的目录下安装即可

Linux

C++依赖版本：

cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

纯python版本：

cd apex
pip install -v --disable-pip-version-check --no-cache-dir ./

Windows

C++依赖版本：

cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .

纯Python版本：

cd apex
pip install -v --no-cache-dir .