（beta）torch_npu.npu.stop_device

接口原型

torch_npu.npu.stop_device(device_id: int) -> None 

功能描述

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

参数说明

“device_id”(int) 需要处理的device id。

输入说明

要确保是一个有效的device。

支持的型号

• Atlas A2 训练系列产品
• Atlas A3 训练系列产品

调用示例

1
2
3
4

import torch
import torch_npu  
torch.npu.set_device(0) 
torch_npu.npu.stop_device(0)