下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

npu.open

功能说明

用于注册NPU设备,当前必须连续调用as_default设置NPU为默认设备。

函数原型

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

参数说明

参数名

输入/输出

描述

device_id

输入

需要初始化的NPU设备ID,如果不传入,默认读取环境变量ASCEND_DEVICE_ID的值,如果ASCEND_DEVICE_ID环境变量未设置,则取值为0。

ASCEND_DEVICE_ID用于指定当前进程所用的昇腾AI处理器逻辑ID,取值范围[0,N-1],设置示例:

export ASCEND_DEVICE_ID=0

返回值

返回npu_device的实例。

调用示例

1
2
import npu_device as npu
npu.open().as_default()
搜索结果
找到“0”个结果

当前产品无相关内容

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