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按照升序配置。
  • 不支持使用DCMI接口时使用该环境变量。DCMI接口说明请参见DCMI API参考
  • 不支持使用HCCN Tool接口时使用该环境变量。HCCN Tool接口说明请参见HCCN Tool 接口参考
  • 不支持使用Profiling功能时使用该环境变量。

该环境变量使用场景:

离线推理程序执行场景。

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

配置示例

使用约束

针对TensorFlow框架网络在昇腾平台执行训练或在线推理的场景,ASCEND_RT_VISIBLE_DEVICES环境变量的优先级高于ASCEND_DEVICE_ID。若同时使用两个环境变量,则ASCEND_DEVICE_ID环境变量处只能设置ASCEND_RT_VISIBLE_DEVICES环境变量中的Device索引值,否则会导致训练或推理应用程序运行异常。

支持的型号

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品