ASCEND_DEVICE_ID
功能描述
指定昇腾AI处理器的逻辑ID。取值范围[0,N-1],默认为0。其中N为当前物理机/虚拟机/容器内的设备总数。
当用户需要将不同的模型通过同一个训练脚本在不同的Device上执行时,可以通过TF Adapter提供的运行参数session_device_id指定昇腾AI处理器的逻辑ID,该种场景下无需配置ASCEND_DEVICE_ID,如果同时配置,则以session_device_id配置的为准。
对于TensorFlow 2.6训练场景,当npu.open接口调用未传入设备ID时,会读取该环境变量。
配置示例
export ASCEND_DEVICE_ID=0
父主题: 基本信息