npu.open

Applicability

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	×
Atlas inference products	×
Atlas training products	√

Description

Registers an NPU device, used in conjunction with as_default to set the NPU as the default device.

Prototype

npu.open(device_id=None).as_default()

Parameters

Parameter	Input/Output	Description
device_id	Input	ID of the NPU device to be initialized. Defaults to the value of the ASCEND_DEVICE_ID environment variable or 0 if the ASCEND_DEVICE_ID environment variable is not set. ASCEND_DEVICE_ID specifies the logical ID of the Ascend AI Processor used by the current process. Value range: [0, N-1]. Example: export ASCEND_DEVICE_ID=0

Parameter

Input/Output

Description

device_id

Input

ID of the NPU device to be initialized. Defaults to the value of the ASCEND_DEVICE_ID environment variable or 0 if the ASCEND_DEVICE_ID environment variable is not set.

ASCEND_DEVICE_ID specifies the logical ID of the Ascend AI Processor used by the current process. Value range: [0, N-1]. Example:

export ASCEND_DEVICE_ID=0

Returns

A npu_device instance.

Example

import npu_device as npu
npu.open().as_default()