昇腾社区首页
中文
注册

(beta)torch_npu.npu.stop_device

须知:[object Object] 本接口为预留接口,暂不支持。

产品支持情况

产品 是否支持
[object Object]Atlas A3 训练系列产品[object Object]
[object Object]Atlas A2 训练系列产品[object Object]

功能说明

停止对应device上的计算,对于没有执行的计算进行清除,后续在此device上执行计算会报错。

函数原型

[object Object]

参数说明

device_id (int):需要处理的device id,确保是一个有效的device。

返回值说明

int

返回值为int,代表执行结果,0表示执行成功,1表示执行失败。

调用示例

[object Object]