ai-for-science-ai4s-basic
Installation
SKILL.md
通用昇腾 NPU 模型迁移 Skill
本 Skill 提供一套通用的、从 CUDA 迁移到昇腾 NPU 的标准化流程, 适用于 PyTorch、TensorFlow、vLLM 等主流框架的模型项目。 迁移完成后,应根据实际跑通步骤生成该模型专属的迁移 Skill,以便复用。
前置条件
执行迁移前确认以下环境就绪:
| 项目 | 要求 |
|---|---|
| 硬件 | Ascend910 系列(至少 1 卡) |
| OS | openEuler / Ubuntu / KylinOS(aarch64 或 x86_64) |
| CANN | ≥ 8.0(推荐 8.2+ 或 8.3.RC1) |
| Python | 3.8 – 3.10(推荐 3.10) |
| PyTorch | 与 CANN 版本匹配(参考华为版本配套表) |
| torch_npu | 与 PyTorch 版本一致 |