设置当前需要使用的NPU device,可反复调用实现不同device的切换。
def set_device(npu_id)
mindietorch.set_device接口由于存在参数校验,在输入非法数据时,可能会抛出异常。故用户必须在try/except语句块内进行调用以及异常处理,防止在使用的过程中出现异常抛出导致程序退出的情况。
参数名称 |
参数类型 |
参数说明 |
---|---|---|
npu_id |
Int |
需要使用的NPU device的ID。 npu_id需要为合法有效的芯片ID,最大支持1024。 |
无