torch_npu::init_npu()

定义文件

torch_npu\csrc\libs\init_npu.h

API接口

torch_npu::init_npu(device_str)
torch_npu::init_npu(device_index)
torch_npu::init_npu(device)

功能描述

初始化NPU设备。

参数说明

• device_str：设备名称。
• device_index：设备index，默认0。
• device：npu设备。