NPU Device
API接口  | 
说明  | 
|---|---|
(beta)torch_npu.npu.is_initialized  | 
Torch_npu提供设备相关的部分接口,具体可参考torch.cuda。  | 
(beta)torch_npu.npu.init  | 
|
(beta)torch_npu.npu.get_device_name  | 
|
(beta)torch_npu.npu.can_device_access_peer  | 
|
(beta)torch_npu.npu.get_device_properties  | 
|
(beta)torch_npu.npu.device_of  | 
|
(beta)torch_npu.npu.current_blas_handle  | 
|
(beta)torch_npu.npu.set_stream  | 
|
(beta)torch_npu.npu.set_sync_debug_mode  | 
|
(beta)torch_npu.npu.get_sync_debug_mode  | 
|
(beta)torch_npu.npu.utilization  | 
|
(beta)torch_npu.npu.get_device_capability  | 
预留接口,暂不支持,接口默认返回None。  | 
父主题: torch_npu