用途:指导把传统CPU/GPU模型迁移到昇腾NPU,覆盖代码适配与环境搭建两个维度。代码层面涉及设备映射、接口适配、数据加载、训练流程与精度处理;环境层面涉及驱动与CANN安装、PyTorch与torch_npu版本匹配、依赖兼容处理。相关子技能包括迁移执行、环境搭建、torch_npu接口参考。
主要能力:分析目标脚本用到的torch接口与第三方库并标记NPU支持情况;按流程完成代码迁移与迭代验证;进行精度、性能对比并输出可复现的迁移报告;可做NPU亲和性调优。
使用方式:按步骤1至步骤5顺序执行——代码分析与接口识别、CPU基线环境搭建、NPU代码迁移、输出结果验证、输出迁移报告。登录NPU服务器优先SSH直连,后台任务用nohup保护,例如nohup python train.py > train.log 2>&1 &,并以tail -f train.log监控。环境搭建须严格参照环境搭建Skill,不自行猜想。
限制:版本号与文档链接会持续更新,执行时须访问昇腾官网实时查询,链接中版本号替换为实际版本;无NPU环境时仍可改代码但须声明未经验证;不修改第三方库源码,避免修改代码仓源码;精度与性能数据必须来自实际运行,不可臆造。