昇腾社区首页
中文
注册

KINETO_USE_DAEMON

功能描述

PyTorch训练场景用于设置是否通过msMonitor nputrace方式开启dynamic_profile采集功能。

配置示例

export KINETO_USE_DAEMON=1

详细使用方式请参见CANN 性能调优工具用户指南》中的“dynamic_profile动态采集”章节

使用约束

  • 脚本不手动添加代码的情况,此环境变量适用于PyTorch训练场景。
  • 脚本内添加dynamic_profile模块后,此环境变量可以在非训练场景使用。例如:
    1
    2
    3
    4
    5
    6
    7
    8
    9
    # 加载dynamic_profile模块
    from torch_npu.profiler import dynamic_profile as dp
    # 设置Profiling配置文件的路径
    dp.init("profiler_config_path")
    
    for step in steps:
    	train_one_step()
    	# 划分step
    	dp.step()
    

支持的型号

  • Atlas 训练系列产品
  • Atlas A2 训练系列产品
  • Atlas A3 训练系列产品