版本配套与环境激活

CANN 的使用有两件「绕不开」的事：版本要配套、环境要激活。本页说明这两点。

1. 版本配套

昇腾软件栈是分层耦合的，自下而上的版本必须相互匹配，否则会出现无法识别设备、算子报错等问题：

驱动 / 固件  ↔  CANN  ↔  torch_npu  ↔  PyTorch

• 驱动 / 固件 ↔ CANN：CANN 依赖匹配的 NPU 驱动与固件版本。
• CANN ↔ torch_npu：torch_npu 的版本号遵循 {PyTorch 版本}-{Ascend 版本} 规则，需与所装 CANN 对应。
• 安装前请查阅官方的版本配套表，确认四者版本一致。

提示

本平台已为你预装并配好匹配版本，一般无需自行安装。若在自有环境中复现，务必先核对版本配套表，这是最常见的踩坑点。

2. 环境检查

在使用框架前，先确认 NPU 设备正常。npu-smi 是昇腾的命令行管理工具，类比 nvidia-smi：

npu-smi info

正常时可看到 NPU 型号、健康状态（Health: OK）、温度与利用率。更完整的排查见 昇腾 NPU 环境检查。

3. 环境激活

CANN 的库与环境变量需要先加载，框架才能找到底层依赖。通常通过执行 CANN 安装目录下的 set_env.sh 完成：

# 路径以实际安装位置为准，常见为 /usr/local/Ascend
source /usr/local/Ascend/ascend-toolkit/set_env.sh

激活后，再 import torch / import torch_npu 即可在 NPU 上运行。

衔接

环境就绪后，进入 AI 框架（PyTorch） 开始在 NPU 上跑代码。