TORCH_NPU_DEVICE_CAPABILITY

功能描述

通过此环境变量可配置torch_npu.npu.get_device_capability()的返回值，仅用于兼容原生PyTorch的torch.cuda.get_device_capability()接口，不代表NPU硬件实际能力。

• 未配置时，torch_npu.npu.get_device_capability()返回值为None。
• 配置时，torch_npu.npu.get_device_capability()返回值为环境变量TORCH_NPU_DEVICE_CAPABILITY的值，配置格式遵循major.minor格式，例如8.0，9.0。

此环境变量默认为未配置。

配置示例

export TORCH_NPU_DEVICE_CAPABILITY=8.0

使用约束

无

支持的型号

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