TensorFlow 2.6.5训练与在线推理场景下,用于开启TF Adapter图Dump功能,该变量需要在import npu_device前设置。
export NPU_DUMP_GRAPH=1
否
该环境变量仅适用于TensorFlow 2.6.5网络在昇腾平台执行训练或在线推理的场景。
Atlas 训练系列产品
Atlas A2 训练系列产品