在 Raspberry Pi 4 上安装 PyTorch

介绍。 本页将指导您在64 位 *** 作系统的 Raspberry Pi 4 上安装PyTorch 1.11.0或更早版本、TorchVision、LibTorch和Caffe2 。 PyTorch 是专门为深度学习开发的软件库。它消耗了大量的 Pi 资源。您可以在 Raspberry Pi 4 上运行 PyTorch，但不要期待奇迹。它运行你的模型，如果不是太复杂的话，但它不能训练新模型。它也不能执行所谓的迁移学习。 PyTorch 在 Python 上运行。有可用的 C++ API，但我们尚未对其进行测试。如果您希望您的 PyTorch 模型在您的 Raspberry Pi 上快速运行，请考虑将您的模型导出到 ONNX。保存为 ONNX 模型后，您可以将其导入 ARM 友好的 C++ 框架，例如 ncnn 或 MNN。它将大大加快您的模型。 如果您的 Raspberry Pi 上有 Ubuntu，则Ubuntu 18.04或20.04有特殊的编译轮。 提示。 我们经常会遇到这样的问题，如果我们有一个带有预装框架和深度学习示例的 Raspberry Pi 4 的 SD 图像。 我们很高兴遵守这一要求。请在我们的GitHub页面上找到专门用于深度学习的完整可用的 Raspberry Pi 4。从我们的 GDrive 站点下载 zip 文件，解压缩并将图像闪存到 16 GB SD 卡上，然后尽情享受吧！

我们只有 Raspberry Pi 64 位 *** 作系统的指南。随着 PyTorch 的发展，在简单的 32 位机器上构建它变得越来越困难。例如，JIT 跟踪 不适用于 32 位 *** 作系统。 此外，今天我们的大多数访问者在他们的 Raspberry Pi 上使用 64 位 *** 作系统来构建他们的深度学习模型，因为它更快、更健壮。

 我们讨论了两种安装，一种带有 Python 3 wheel 。另一种方法是从头开始构建。不幸的是，没有可用于 Raspberry Pi 4 的官方 pip3 wheel 。但是，为了您的方便，我们创建了这些轮并将它们放在GitHub 上。 版本检查。 在 Raspberry 64 位 *** 作系统上安装 PyTorch 之前，请检查您的 *** 作系统。运行命令 uname -a 并使用下面的屏幕转储验证您的版本。

您还需要使用命令gcc -v检查您的 C++ 编译器版本。它还必须是aarch64-linux-gnu版本，如屏幕截图所示。如果您有 64 位 *** 作系统，但您的 gcc 版本与上面给出的不同，请使用最新版本重新安装整个 *** 作系统。该指南可在此处找到：在 Raspberry Pi 4 上安装 64 位 *** 作系统。因为我们要构建库，所以需要一个 64 位 C++ 编译器。即使你使用 Python 轮子，gcc 也会在幕后被调用。 同样重要的是，根据我们的说明安装后的 zram 交换大小超过 3 GB。只要您不从头开始构建 PyTorch，Less 就没有问题。 wheel安装 PyTorch 由 Ninja 构建。完成整个构建大约需要 3 个小时。我们已经在我们的 GitHub 页面上发布了这些wheel。随意使用这些。完成所有繁琐的工作后，现在只需几分钟即可在 Nano 上安装 PyTorch。对于顽固分子，完整的程序将在本手册后面介绍。 整个过程如下。wheel太大，无法在 GitHub 上存储，因此使用了 Google 驱动器。请确保您安装了最新的 pip3 和 python3 版本，否则 pip 可能会出现消息“ .whl is not a supported wheel on this platform ”。 Python检查。 检查您的 Python3 版本。每个版本都需要一个独特的wheel。树莓派 64 位 *** 作系统使用 Python 3.7.3。所以需要下载torch-1.XY- cp37 - cp37m -linux_aarch64.whl。如果你有最新的 Raspberry Pi Bullseye OS，你的 Python 版本是 3.9.2。 毫无疑问，Python 版本会随着时间的推移而升级，您将需要一个不同的轮子。查看所有wheel的GitHub 页面。 PyTorch1.11.0版本安装

# get a fresh start
sudo apt-get update
sudo apt-get upgrade
#install the dependencies (if not already onboard)
sudo apt-get install python3-pip libjpeg-dev libopenblas-dev libopenmpi-dev libomp-dev
#above 58.3.0 you get version issues
sudo -H pip3 install setuptools==58.3.0
sudo -H pip3 install Cython
#install gdown to download from Google drive
sudo -H pip3 install gdown

Buster OS
#download the wheel
gdown https://drive.google.com/uc?id=1gAxP9q94pMeHQ1XOvLHqjEcmgyxjlY_R
# install PyTorch 1.11.0
sudo -H pip3 install torch-1.11.0a0+gitbc2c6ed-cp37-cp37m-linux_aarch64.whl
# clean up
rm torch-1.11.0a0+gitbc2c6ed-cp37-cp37m-linux_aarch64.whl

or Bullseye OS
# download the wheel
gdown https://drive.google.com/uc?id=1ilCdwQX7bq72OW2WF26Og90OpqFX5g_-
# install PyTorch 1.11.0
sudo -H pip3 install torch-1.11.0a0+gitbc2c6ed-cp39-cp39-linux_aarch64.whl
# clean up
rm torch-1.11.0a0+gitbc2c6ed-cp39-cp39-linux_aarch64.whl

PyTorch1.10.0版本安装

# get a fresh start
sudo apt-get update
sudo apt-get upgrade
# install the dependencies (if not already onboard)
sudo apt-get install python3-pip libjpeg-dev libopenblas-dev libopenmpi-dev libomp-dev
# above 58.3.0 you get version issues
sudo -H pip3 install setuptools==58.3.0
sudo -H pip3 install Cython
# install gdown to download from Google drive
sudo -H pip3 install gdown

Buster OS
# download the wheel
gdown https://drive.google.com/uc?id=1y-X1UGC43xUfnycJcuDp7tlyFi8QHxvs
# install PyTorch 1.10.0
sudo -H pip3 install torch-1.10.0a0+git36449ea-cp37-cp37m-linux_aarch64.whl
# clean up
rm torch-1.10.0a0+git36449ea-cp37-cp37m-linux_aarch64.whl

or Bullseye OS
# download the wheel
gdown https://drive.google.com/uc?id=10zJmU1f7PoxN8ARZol4mtbwok4G23XKZ
# install PyTorch 1.10.0
sudo -H pip3 install torch-1.10.0a0+git36449ea-cp39-cp39-linux_aarch64.whl
# clean up
rm torch-1.10.0a0+git36449ea-cp39-cp39-linux_aarch64.whl

PyTorch1.9.0版本安装

# get a fresh start
sudo apt-get update
sudo apt-get upgrade
# install the dependencies (if not already onboard)
sudo apt-get install python3-pip libjpeg-dev libopenblas-dev libopenmpi-dev libomp-dev
# above 58.3.0 you get version issues
sudo -H pip3 install setuptools==58.3.0
sudo -H pip3 install Cython
# install gdown to download from Google drive
sudo -H pip3 install gdown

Buster OS
# download the wheel
gdown https://drive.google.com/uc?id=18KztAf3W_kqFYP2f9uqLj3HBdhoMvVe4
# install PyTorch 1.9.0
sudo -H pip3 install torch-1.9.0a0+gitd69c22d-cp37-cp37m-linux_aarch64.whl
# clean up
rm torch-1.9.0a0+gitd69c22d-cp37-cp37m-linux_aarch64.whl

or Bullseye OS
# download the wheel
gdown https://drive.google.com/uc?id=1p9Sp5YmCJwIkmQIWkpuIkE4NgeYPOr-U
# install PyTorch 1.9.0
sudo -H pip3 install torch-1.9.0a0+gitd69c22d-cp39-cp39-linux_aarch64.whl
# clean up
rm torch-1.9.0a0+gitd69c22d-cp39-cp39-linux_aarch64.whl

成功安装后，您可以使用以下命令检查 PyTorch。