昇腾社区首页
中文
注册
开发者
下载

采集Profiling数据

# 安装插桩包
pip install pytorch-2.1.0/dist/torch-*.whl --force-reinstall --no-deps
pip install torch_npu/dist/torch_npu-*.whl --force-reinstall --no-deps

# 设置关键环境变量
export OMP_PROC_BIND=false
export LLVM_PROFILE_FILE=/tmp/profile/default_%m.profraw  # 确保/tmp/profile目录为空

# 执行实际训练任务采集性能数据,例如: bash run_model.sh
...