ASCEND_DEVICE_ID

功能描述

指定芯片的逻辑ID。取值范围[0,N-1],默认为0。其中N为当前物理机/虚拟机/容器内的设备总数。

当用户需要将不同的模型通过同一个训练脚本在不同的Device上执行时,可以通过TF Adapter提供的运行参数session_device_id指定芯片的逻辑ID,该种场景下无需配置ASCEND_DEVICE_ID,如果同时配置,则以session_device_id配置的为准。

对于TensorFlow 2.6训练场景,当npu.open接口调用未传入设备ID时,会读取该环境变量。

配置示例

export ASCEND_DEVICE_ID=0