昇腾社区首页
中文
注册

概述

msleaks工具是基于昇腾AI处理器的内存检测工具,用于模型训练与推理过程中的内存问题定位。msleaks工具提供Step内和Step间的内存异常检测能力,包括Step内内存泄漏分析和Step间内存对比分析。

当前msleaks工具仅支持PyTorch 7.0.0及之后版本的深度学习模型。

功能特性

msleaks工具目前已支持的功能如表1

表1 msleaks工具功能特性

功能

使用场景及说明

Step内内存泄漏分析

针对内存长时间未释放、内存泄漏等问题,需要进行Step内内存分析的情况,msleaks工具提供Step内内存泄漏分析和kernelLaunch粒度的内存变化分析功能,进行告警定位与分析。

Step间内存对比分析

当两个Step内存使用存在差异,可能造成内存使用过多,甚至OOM(Out of Memory,内存溢出)的问题,则需要使用msleaks工具的Step间内存对比分析功能来定位并分析问题。