通过tensorboard查看内存使用情况,内存分析可参见
《PyTorch模型迁移和训练指南》中“性能调优 > Profiling数据采集与分析 > (推荐)Ascend PyTorch Porfiler数据采集与分析”章节中“性能分析(Memory View)”部分。显示效果如
图1所示。
图1 界面显示图
- 绿色折线(Reserved)表示内存池向驱动申请的内存总量。
- 蓝色折线(Allocated)表示内存池当前被使用的内存总量。
左键选中一个区域,可以放大显示。右键返回全图模式。下方的表格列出了申请内存的算子,可以按照大小、申请时间、或释放时间来排序。可以通过表格左上方的搜索栏搜索相关算子,也可以通过表格右上方的滑轮拖动控制表格显示所选区域申请特定大小内存的算子。