ASCEND_RT_VISIBLE_DEVICES

功能描述

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

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

• 设置ASCEND_RT_VISIBLE_DEVICES=1，则表示当前进程仅可使用device ID为1的device，获取到的device数量为1，此时索引值0对应的device ID为1。
• 设置ASCEND_RT_VISIBLE_DEVICES=1,2,3，则表示当前进程仅可使用device ID为1、2、3的三个device，获取到的device数量为3，此时索引值0对应的device ID为1。

该环境变量使用场景：

离线推理程序执行场景。

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

配置示例

export ASCEND_RT_VISIBLE_DEVICES=1

是否必选

否

使用约束

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

支持的型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas 推理系列产品（Ascend 310P处理器）

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

Atlas 200I/500 A2推理产品