mindietorch.set_device

函数功能

设置当前需要使用的NPU device,可反复调用实现不同device的切换。

函数原型

def set_device(npu_id)

约束说明

mindietorch.set_device接口由于存在参数校验,在输入非法数据时,可能会抛出异常。故用户必须在try/except语句块内进行调用以及异常处理,防止在使用的过程中出现异常抛出导致程序退出的情况。

参数说明

参数名称

参数类型

参数说明

npu_id

Int

需要使用的NPU device的ID。

npu_id需要为合法有效的芯片ID,最大支持1024。

返回值说明