开发者
资源
[object Object][object Object]

torch_npu\csrc\libs\init_npu.h

[object Object]
[object Object]
[object Object]

获取当前NPU设备,返回值类型DeviceIndex,与c10::DeviceIndex c10::cuda::current_device()相同。

[object Object]
  • [object Object]Atlas 训练系列产品[object Object]
  • [object Object]Atlas A2 训练系列产品[object Object]
  • [object Object]Atlas A3 训练系列产品[object Object]
  • [object Object]Atlas 推理系列产品[object Object]