Pytorch使用手册—Raspberry Pi 4 上的实时推理(30 FPS!)(专题三十六)
PyTorch 原生支持 Raspberry Pi 4。本教程将指导您如何设置 Raspberry Pi 4 以运行 PyTorch,并在 CPU上以 30 FPS 以上的速度实时运行 MobileNet v2 分类模型。
该教程在 Raspberry Pi 4 Model B 4GB 上进行了测试,但也应适用于2GB 版本,以及 Raspberry Pi 3B,不过性能会有所降低。
前提条件
为了跟随本教程,您需要一台 Raspberry Pi 4、一台相机以及其他标准配件。
- Raspberry Pi 4 Model B 2GB+
- Raspberry Pi 摄像头模块
- 散热片和风扇(可选,但建议使用)
- 5V 3A USB-C 电源适配器
- SD 卡(至少 8GB)
- SD 卡读写器
Raspberry Pi 4 设置
PyTorch 只提供 Arm 64 位(aarch64)的 pip 包,因此您需要在 Raspberry Pi 上安装 64 位版本的操作系统。
您可