一、边缘AI与GPU概览

大家好,我是你们这门课的主讲。在正式开始动手部署之前,我觉得有必要先聊聊“为什么”。为什么边缘计算需要GPU?为什么不是CPU硬扛?

我最早接触边缘AI,是在一个工业质检项目里。客户要求在产线旁边放一台小盒子,实时检测产品缺陷。当时我第一反应是“上服务器”,结果被现场工程师怼了——网络延迟、带宽成本、数据隐私,全是坑。从那以后,我算是彻底明白了:边缘计算不是把云搬下来,而是另一种玩法。

1.1 边缘计算概念:为什么我们需要它?

边缘计算,说白了就是把计算能力从云端下沉到数据产生的地方。你想想看,摄像头拍下一张图,如果传到云端再返回结果,一来一回可能几百毫秒。对于自动驾驶、工业质检这种场景,黄花菜都凉了。

边缘计算的核心价值就三点:

  • 低延迟:数据就地处理,毫秒级响应
  • 省带宽:不用把所有原始数据都往云上送
  • 保隐私:敏感数据不出设备,合规压力小

我个人习惯把边缘设备分成三类:

  • 轻量级:树莓派、Jetson Nano这类,功耗几瓦到十几瓦
  • 中量级:Jetson Xavier NX、Intel NUC,能跑中等模型
  • 重量级:Jetson AGX Orin、边缘服务器,接近桌面级性能
我的经验:别一上来就选最贵的。我在一个项目中用了AGX Orin,结果发现模型根本跑不满,纯属浪费。先评估你的模型复杂度,再选设备。

1.2 AI推理需求:训练和推理是两码事

很多人搞混训练和推理。训练是“学习”,推理是“应用”。在边缘设备上,我们只做推理。

推理有几个特点:

  • 实时性:比如目标检测,30FPS是底线
  • 低功耗:边缘设备往往电池供电或者散热有限
  • 模型压缩:大模型跑不动,得量化、剪枝、蒸馏

为什么会这样?因为边缘设备的算力天花板就在那。你想想看,一个Jetson Nano的算力只有472 GFLOPS,而一块RTX 3090是35 TFLOPS,差了将近80倍。所以,别指望在边缘设备上跑GPT-3,不现实。

我曾经在一个项目中,客户非要把ResNet-152部署到Jetson Nano上。结果呢?推理一次要3秒,根本没法用。后来换成MobileNetV3,量化到INT8,推理时间降到30毫秒。嗯,这里要注意:模型选型决定了项目成败。

1.3 GPU在边缘的角色:不只是加速

GPU在边缘设备上,不只是“跑得快”那么简单。它有几个关键角色:

  • 并行计算引擎:CNN里的卷积操作,GPU天生适合
  • 视频解码器:很多边缘GPU自带硬件解码,比如Jetson的NVDEC
  • Tensor Core:NVIDIA的Tensor Core能加速INT8/FP16推理,吞吐量翻倍

我建议你把GPU看作一个“专用加速器”,而不是通用CPU。它的强项是矩阵运算,弱项是逻辑控制。所以,部署时要尽量把计算密集型操作交给GPU,控制逻辑留给CPU。

关键点:边缘GPU的显存通常很小(4GB-16GB),模型加载时要格外注意显存占用。我见过有人把FP32模型直接塞进去,结果OOM(显存溢出),项目直接卡住。

1.4 主流边缘GPU平台对比

目前市面上主流的边缘GPU平台,我列个表给你看:

平台 算力 显存 功耗 典型场景
Jetson Nano 472 GFLOPS 4GB 10W 入门级AI、教育
Jetson TX2 NX 1.33 TFLOPS 4GB 15W 轻量工业、无人机
Jetson Xavier NX 21 TOPS (INT8) 8GB 15W 中等AI、机器人
Jetson AGX Orin 275 TOPS (INT8) 32GB 60W 高级AI、自动驾驶
Intel NUC + Arc 约8 TFLOPS 6GB 65W 边缘服务器、视觉

我个人最常用的是Jetson系列,原因很简单:生态成熟。NVIDIA提供了JetPack SDK,包含了CUDA、cuDNN、TensorRT,开箱即用。Intel的Arc显卡虽然性能不错,但软件生态还在追赶。

避坑指南:我曾经在一个项目中选了Jetson TX2,结果发现它的CPU是ARM架构,很多x86的库没法直接用。后来花了两天时间交叉编译,才把依赖搞定。所以,选平台时一定要确认软件兼容性。

1.5 本章小结

好了,这一章我们聊了边缘计算的概念、AI推理的需求、GPU的角色,以及主流平台的对比。说白了,边缘AI部署不是把云端的模型直接搬下来,而是要针对硬件做适配。

下一章,我们会深入TensorRT的优化技巧。到时候我会手把手教你如何把模型量化到INT8,以及如何避免那些我踩过的坑。嗯,先到这,我们下章见。