【备忘录】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。版本选择示意图如下:
【备忘录】torch171+cu110配置apex_第1张图片
必须先切换到指定版本下再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 .

你可能感兴趣的:(python,开发语言)