跳转至

Ascend Pytorch Profiler

Ascend Pytorch Profiler 是面向 PyTorch 训练与在线推理场景的性能数据采集工具,支持采集框架侧算子信息、CANN 软件栈数据、底层 NPU 算子信息以及显存占用信息,并可衔接自动解析、离线解析、MindStudio Insight 和 msprof-analyze 进行后续分析。

文档导读

典型流程

  1. 完成环境准备并选择采集方式。
  2. 在脚本中接入 torch_npu.profiler.profile、或使用dynamic_profile动态采集。
  3. 执行训练或在线推理任务,生成性能数据。
  4. 选择自动解析或离线解析。
  5. 使用 MindStudio Insight 或 msprof-analyze 查看和分析结果。