在Raspberry Pi 3 Model B Rev 1.2上安装PyTorch失败求助

问题原因 #

树莓派3B Rev1.2基于ARMv7架构，PyTorch官方CPU源没有适配该架构的预编译包，因此直接使用官方命令会出现版本匹配错误。

解决步骤 #

1. 优先使用piwheels源安装 #

piwheels提供了适配树莓派ARM架构的PyTorch预编译包，执行以下命令：

pip3 install torch torchvision torchaudio --extra-index-url https://www.piwheels.org/simple

如果你的虚拟环境中Python版本为3.8-3.10，这个方法大概率可行。

2. 手动下载对应版本安装 #

若上述命令失败，可手动下载匹配Python版本和ARMv7架构的whl包：

• 找到对应版本的torch、torchvision、torchaudio包（比如针对Python3.9的torch-2.0.0-cp39-none-linux_armv7l.whl）
• 本地安装：

# 先安装torch
pip3 install ./torch-xxx-cp3x-none-linux_armv7l.whl
# 再安装对应版本的torchvision和torchaudio
pip3 install ./torchvision-xxx-cp3x-none-linux_armv7l.whl ./torchaudio-xxx-cp3x-none-linux_armv7l.whl

3. 编译安装（备选，耗时较长） #

若以上方法均无效，只能手动编译适配ARMv7的PyTorch：

1. 安装编译依赖：

sudo apt-get update && sudo apt-get install build-essential cmake git python3-dev python3-numpy

2. 克隆指定版本的PyTorch仓库（以v2.0.0为例）：

git clone --depth 1 --branch v2.0.0 https://github.com/pytorch/pytorch.git
cd pytorch
git submodule update --init --recursive --depth 1

3. 设置编译环境变量（关闭非必要特性以加快编译）：

export NO_CUDA=1
export NO_DISTRIBUTED=1
export NO_MKLDNN=1
export NO_NNPACK=1
export NO_QNNPACK=1

4. 开始编译并安装：

python3 setup.py install

注意：树莓派3B性能有限，编译过程可能持续3-6小时，期间需确保设备稳定供电。

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