2、嵌入式平台选型:主流嵌入式AI平台对比
做嵌入式AI部署,第一步就是选平台。
说实话,我见过不少朋友,模型调得漂漂亮亮,结果一上板子跑不起来。要么内存不够,要么算力跟不上。所以这一章,咱们把市面上最主流的三个平台——Jetson Nano、RK3588、树莓派4B——掰开揉碎了聊一聊。
2.1 三大平台核心参数对比
先看一张表,心里有个底。
| 参数项 | Jetson Nano | RK3588 | 树莓派4B |
|---|---|---|---|
| AI算力 | 472 GFLOPS (FP16) | 6 TOPS (INT8) | ≈ 0.1 TOPS (CPU) |
| CPU | 4核 Cortex-A57 | 4核 A76 + 4核 A55 | 4核 Cortex-A72 |
| GPU | 128核 Maxwell | Mali-G610 MP4 | VideoCore VI |
| NPU | 无 | 6 TOPS NPU | 无 |
| 内存 | 4GB LPDDR4 | 8GB/16GB LPDDR4X | 2GB/4GB/8GB LPDDR4 |
| 功耗 | 5W-10W | 8W-15W | 3W-7W |
| 价格 | 约 800-1000 元 | 约 1200-1800 元 | 约 300-600 元 |
嗯,光看数字可能有点抽象。我一个个说。
2.2 Jetson Nano:老牌劲旅,生态成熟
Jetson Nano 是我最早接触的嵌入式AI平台。说实话,它不算新了,但胜在稳定。
它的 472 GFLOPS 算力,跑 YOLOv5s 的 FP16 模型,大概能到 20-25 FPS。这个帧率做实时检测,勉强够用。
不过有个坑——它没有 NPU。所有AI计算全靠GPU。这意味着你跑模型的时候,GPU 占用率经常冲到 95% 以上。我有个项目,同时跑检测和显示,结果画面卡成幻灯片。后来把显示任务扔给 CPU,才勉强稳住。
2.3 RK3588:国产新秀,性价比之选
RK3588 是瑞芯微的旗舰芯片。我第一次拿到开发板时,第一反应是——这玩意儿真能打。
6 TOPS 的 NPU 算力,比 Jetson Nano 的 GPU 算力强了不止一个量级。而且它还有 4 个 A76 大核,跑预处理和后处理非常快。
我做过一个测试:用 RK3588 的 NPU 跑 YOLOv5s 的 INT8 模型,帧率能到 40-50 FPS。这个成绩,说实话,让我有点意外。
不过它也有短板。生态不如 Jetson 成熟。你遇到问题,网上搜到的资料可能不多。我曾经卡在一个算子兼容性问题上,折腾了两天才搞定。
2.4 树莓派4B:入门可以,实战勉强
树莓派4B,很多人的第一块开发板。我也不例外。
但说实话,用它跑 YOLO,纯属「用爱发电」。它的 CPU 算力只有 0.1 TOPS 左右,跑 YOLOv5s 的 FP32 模型,帧率大概 2-3 FPS。嗯,就是看幻灯片的感觉。
那它适合什么?
- 学习 Linux 和嵌入式基础
- 跑一些轻量级的分类模型
- 做原型验证,不追求实时性
我有个学生,非要用树莓派做实时口罩检测。我劝了他三次,他不听。最后跑出来 1.5 FPS,他自己放弃了。
2.5 算力与内存需求分析
选平台,说白了就是看两个东西:算力和内存。
算力方面:
- YOLOv5n / YOLOv8n:需要 0.5-1 TOPS,树莓派勉强能跑,但帧率很低
- YOLOv5s / YOLOv8s:需要 2-4 TOPS,Jetson Nano 和 RK3588 都能胜任
- YOLOv5m / YOLOv8m:需要 6-10 TOPS,RK3588 勉强,Jetson Nano 吃力
- YOLOv5l 以上:需要 10+ TOPS,这两个平台都别想了
内存方面:
- YOLOv5n:约 200-300 MB
- YOLOv5s:约 500-700 MB
- YOLOv5m:约 1.2-1.5 GB
- YOLOv5l:约 2.5-3 GB
你想想看,Jetson Nano 只有 4GB 内存,系统占掉 1GB,再跑个 YOLOv5m,内存就快满了。所以我一般只推荐在它上面跑 YOLOv5s 或 YOLOv8n。
2.6 选型建议
说了这么多,到底怎么选?我直接给结论。
- 预算有限,入门学习: 树莓派4B(300-600元)
- 项目实战,追求稳定: Jetson Nano(800-1000元)
- 性能优先,性价比高: RK3588(1200-1800元)
- 量产产品,成本敏感: RK3588(芯片价格更低)
我个人习惯是:
- 做原型验证,用 Jetson Nano。生态好,遇到问题好解决。
- 做量产产品,用 RK3588。性能强,成本低。
- 树莓派?嗯,我拿它当 Linux 服务器用,跑跑脚本还行。
最后说一句:没有最好的平台,只有最合适的平台。你手头有什么项目,就选什么板子。别为了省钱买树莓派,结果项目做不成,反而更浪费钱。
下一章,咱们聊聊怎么在选好的板子上搭建环境。到时候我会手把手教你装系统、配驱动、跑第一个 YOLO 模型。