安装与版本配套

本平台已预装

本平台镜像通常已预装好匹配版本的 PyTorch 与 torch_npu，一般无需自行安装。本页供在自有环境复现时参考。

1. 版本配套规则

torch_npu 的版本号遵循 {PyTorch 版本}-{Ascend 版本} 的命名规则，安装时三者必须匹配：

PyTorch  ↔  torch_npu  ↔  CANN

• PyTorch 与 torch_npu 的主版本要一致（如 PyTorch 2.1 对应 torch_npu 2.1.x）。
• torch_npu 又要与所装的 CANN 版本对应。
• 安装前请查阅官方版本配套表确认。

2. 安装步骤

# 1. 安装对应版本的原生 PyTorch（CPU 版即可，算力由 NPU 提供）
pip install torch==2.1.0

# 2. 安装匹配的 torch_npu
pip install torch-npu==2.1.0

# 依赖项按官方要求补齐（如 apex、torchvision 等按需安装）

具体版本号以官方版本配套表与本平台环境为准，上面仅为示例。

3. 安装验证

安装并激活 CANN 环境后，运行：

import torch
import torch_npu

print(torch.__version__)
print(torch_npu.__version__)
print(torch.npu.is_available())      # 预期输出 True
print(torch.npu.device_count())      # NPU 数量

若 torch.npu.is_available() 返回 True，说明 PyTorch 已能正常使用昇腾 NPU。下一步见 代码迁移。