专栏底层软件 BPU 推理输出持续增长内存

BPU 推理输出持续增长内存

已解决
默认532482025-07-28
32
2

在使用地平线 Python SDK (pyeasy_dnn) 运行 YOLOv5 模型时,

关键代码:

from hobot_dnn import pyeasy_dnn as dnn

models = dnn.load('xxx.bin') results = models[0].forward(nv12_data) 在使用地平线 Python SDK (pyeasy_dnn) 运行 YOLOv5 模型时,调用 models[0].forward(nv12_data) 推理图像,每帧执行一次,程序会持续增长内存,最终导致 OOM。

已确认情况: 每一帧都会调用 forward() 方法获取推理结果; SDK 中未找到的资源释放函数; 使用 del outputs 并不会阻止内存持续增长; 确认内存泄漏位于 BPU 推理输出阶段。

请求: 是否有推荐的方式在 Python 中正确释放 forward() 结果返回的资源?是否 SDK 有新版本提供了释放函数或上下文管理机制?

底层软件
技术深度解析
评论2
0/1000
  • Jade-self
    Lv.5

    建议该问题在地瓜社区询问,智驾社区没有python api推理的内容:https://developer.d-robotics.cc/

    2025-07-28
    0
    0
  • DR_KAN
    Lv.4

    如果是征程系列芯片,并且涉及到easydnn,那么建议走项目支持哈~

    2025-07-28
    0
    0