ASCEND_RT_VISIBLE_DEVICES

功能描述

指定哪些Device对当前进程可见,支持一次指定一个或多个Device ID。通过该环境变量,可实现不修改应用程序即可调整所用Device的功能。

其中Device ID为昇腾AI处理器的逻辑ID,例如可用Device数量为8,Device ID分别为:0、1、2、3、4、5、6、7,使用场景说明如下:

  • 该环境变量为试用环境变量,后续版本可能存在变更,不支持应用于商用产品中。
  • 当前版本仅支持指定的Device ID按照升序配置。

该环境变量使用场景:

离线推理程序执行场景。

TensorFlow/PyTorch等框架网络在昇腾平台执行训练或在线推理的场景。

配置示例

export ASCEND_RT_VISIBLE_DEVICES=1

是否必选

使用约束

针对TensorFlow框架网络在昇腾平台执行训练或在线推理的场景,ASCEND_RT_VISIBLE_DEVICES环境变量的优先级高于ASCEND_DEVICE_ID。

支持的型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas 推理系列产品

Atlas A2训练系列产品/Atlas 800I A2推理产品

Atlas 200I/500 A2推理产品