本文介绍如何在树莓派4B中安装pytorch。

在装pytorch之前，需要强调几个地方，这也是我踩过的坑，树莓派上尽量按照python3.7或python3.9，不要安装最新的(因为很可能没有对应的pytorch),python3.8的当然也可以找到对应版本的pytorch，但对应的opencv不好编译，总是出错。

所以我这里推荐的环境为：python3.9+opencv4.5.2.54+torch1.11.0+numpy1.26.4

先以我推荐的环境为例(文末我会放32位系统python3.7环境安装说明)

1.python3.9环境下torch安装

查看自己树莓派环境：

Linux raspberrypi 6.1.21-v8+ #1642 SMP PREEMPT Mon Apr  3 17:24:16 BST 2023 aarch64 GNU/Linux

可以看我的系统是64位的(推荐)即aarch64,同时我的系统OS安装的是树莓派的Bullseye的，OS分为（Bullseye / Bookworm），官方现在主推的是Bookworm的，但是Qengineering推出的pytorch明确说明了暂时不支持python3.11和Bookworm，所以这也是为什么建议OS为Bullseye的，虽然Bookworm还有其他社区可以安装pytorch，但发现其他库会有问题，例如opencv 4.X版本。

所以这里推荐的OS为Bullseye+pytorch.

树莓派系统安装(64位Bullseye版本)

如果你系统本来就是这个可以忽略这一步！

Bullseye版本的OS安装(这里也建议不要用树莓派官网最新的Imager安装工具，对于自己下载的Bullseye无法设置用户名和密码，非常的坑！！！！用我后面推荐的老版本的Image反而更加友好)

百度网盘：

https://pan.baidu.com/s/1Otgt6lb0rn50n4MpyvuD4A?pwd=yypn 提取码: yypn

这个网盘里有个python3.9的里面有对应的os镜像(后缀名为img.xz)和树莓派官方提供的Imager安装工具 Imager1.7.1

下载Imager1.7.1和2023-05-03-raspios-bullseye-arm64.img.xz(这个镜像安装完默认是python3.9)，打开镜像安装工具，如下：

(如果你是安装Bullseye或者老版本，一定不要更新！！)

img.xz(不用解压)，选择你树莓派的SD卡，在烧录的时候提前设置好无线和用户名！

torch安装

完成OS系统的烧录，新建虚拟环境，命令如下(我这里用的是MobaXterm)：

python3 -m venv ~/python39

激活你的虚拟环境：

source ~/python39/bin/activate

然后你的终端界面显示如下，说明你激活成功，也就是前面有你虚拟环境的名字

(python39) pi@raspberrypi:

将网盘中对应python3.9的torch和torchvision下载下来用pip命令安装即可。

在安装opencv之前需要先安装numpy，这里的numpy推荐1.26.4，然后安装网盘中的opencv

如果想要其他的whl包可以从这里下载，比较快：

https://pypi.org/simple/

输入命令：

只要torch和opencv正常就没什么问题了！你就可以正常使用了。opencv一定要试！！

(python39) pi@raspberrypi:~/Videos $ python3
Python 3.9.2 (default, Feb 28 2021, 17:03:44)
[GCC 10.2.1 20210110] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import cv2

2.python3.7环境下torch安装

和上面不一样，这里说的是armv7，python3.7安装torch

查看自己树莓派的环境：

uname -a

pi@raspberrypi:~ $ uname -a
Linux raspberrypi 4.19.118-v7l+ #1311 SMP Mon Apr 27 14:26:42 BST 2020 armv7l GNU/Linux
 

这里的架构是armv7的。

安装pytorch，可以直接用我百度网盘中的，我这里的python版本是3.7的，网盘中提供pytorch1.8.1和pytorch1.3.0版本的
链接: https://pan.baidu.com/s/1Otgt6lb0rn50n4MpyvuD4A?pwd=yypn 提取码: yypn 

其他版本的pytorch:

https://torch.kmtea.eu/whl/stable.html

使用pip3 install 命令进行安装即可。

如果报错：

RuntimeError: module compiled against API version 0xe but this version of numpy is 0xd

这是因为numpy版本太低了，可以尝试用numpy 1.21.4版本(我把1.21.4的包也放在百度网盘中)。

如果需要其他版本或者其他安装包，可以离线下载，注意架构！https://www.piwheels.org/packages.html

测试pytorch安装是否成功

python3

>>> import torch
>>>

没有报错就说明安装成功了。

---