OOM_SNAPSHOT_PATH

功能描述

通过此环境变量可配置在网络发生内存不足报错时，内存数据保存路径。

• 未配置时，内存数据默认保存至当前路径。
• 配置时，内存数据保存至指定路径。

此环境变量默认为未配置。

必须与OOM_SNAPSHOT_ENABLE配套使用。

配置示例

export OOM_SNAPSHOT_PATH="/home/usr/"

使用约束

此环境变量仅适用于基于PyTorch框架构建的神经网络，且使用PyTorch内存机制管理内存的场景。

支持的型号

• Atlas 训练系列产品
• Atlas A2 训练系列产品
• Atlas A3 训练系列产品
• Atlas 800I A2 推理产品