准备工具
Ascend PyTorch Profiler
参见《PyTorch模型迁移和训练指南》中“性能调优 > Profiling数据采集及分析 > (推荐)Ascend PyTorch Profiler数据采集与分析”,Ascend PyTorch Profiler接口可全面采集PyTorch训练场景下的性能数据,主要包括PyTorch层算子信息、CANN层算子信息、底层NPU算子信息、以及算子内存占用信息等,可以全方位分析PyTorch训练时的性能状态。
父主题: 性能采集与分析工具使用