文档
注册

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
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词