OOM_SNAPSHOT_PATH
功能描述
此环境变量用于配置在网络发生oom时内存数据保存路径。
- 未配置时,内存数据默认保存至当前路径。
- 配置时,内存数据保存至指定路径。
此环境变量默认为未配置。
必须与OOM_SNAPSHOT_ENABLE配套使用。
配置示例
export OOM_SNAPSHOT_PATH="/home/usr/"
使用约束
此环境变量仅适用于PyTorch网络,且使用PyTorch内存机制管理内存的场景。
支持的型号
Atlas 训练系列产品
Atlas A2 训练系列产品