无法通过pip和conda安装PyTorch 1.7.1及对应依赖的问题求助

问题根源 #

PyTorch 1.7.1发布于2020年10月，而Python3.10正式推出是在2021年10月，官方并未为PyTorch1.7.1制作适配Python3.10的预编译包，这就是pip/conda安装时提示“无匹配发行版”的直接原因。

可行解决方案 #

方案1：创建兼容Python版本的conda环境（推荐） #

PyTorch1.7.1支持Python3.6至3.9，优先用以下步骤创建适配环境：

1. 创建指定Python版本的conda环境：

   conda create -n torch171_env python=3.8
   

2. 激活环境：

   conda activate torch171_env
   

3. 执行官方安装命令（此时会匹配到对应预编译包）：
   • Conda安装：

     conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cpuonly -c pytorch
     

   • Pip安装：

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

方案2：在Python3.10环境下从源码编译 #

如果必须保留Python3.10环境，只能手动编译源码：

1. 安装编译依赖：

   conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses
   

2. 拉取PyTorch1.7.1源码并初始化子模块：

   git clone --branch v1.7.1 https://github.com/pytorch/pytorch.git
   cd pytorch
   git submodule update --init --recursive
   

3. 设置编译路径并安装PyTorch（CPU版本）：

   export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
   python setup.py install
   

4. 编译安装torchaudio0.7.2：

   git clone --branch v0.7.2 https://github.com/pytorch/audio.git
   cd audio
   python setup.py install
   

内容的提问来源于stack exchange，提问作者falconx999