ascend-ai-coding Ascend NPU Driver & Firmware Auto-Install
该技能为提供端到端自动化的NPU驱动和固件安装能力，覆盖从安装包校验到驱动生效验证的全流程，无需手动分步操作，严格遵循昇腾官方安装规范，适配主流Linux发行版。
核心能力
- 基于官方正则自动提取指定文件夹内的驱动/固件包，强制校验包数量唯一性，仅允许1个驱动包+1个固件包；
- 安装包可执行权限按需赋权，无权限时自动执行
chmod +x并二次校验赋权结果，避免权限问题导致安装失败； - Python+Shell双重包校验，提前验证包格式、路径、文件有效性，拦截无效安装包；
- 系统依赖先验后装
- 严格按昇腾官方先驱动后固件顺序安装，安装后提供交互式重启选项+官方
npu-smi原生命令验证驱动状态。
前置准备
- 该版本无需校验系内核，直接进行部署即可
1. 脚本文件准备
将核心脚本check_package.py（Python包校验）和install_npu_driver.sh（Shell主安装）放在同一目录，本技能的根目录建议命名为ascend-npu-driver-install，与name字段保持一致。
2. 安装包要求
指定的安装包文件夹内仅存放1个符合昇腾官方命名格式的驱动.run包和1个固件.run包，无其他无关文件，包名格式严格遵循：
- 驱动包：
Ascend-hdk-<chip_type>-npu-driver_<version>_linux-<arch>.run - 固件包：
Ascend-hdk-<chip_type>-npu-firmware_<version>.run
3. 系统环境要求
- 权限：必须拥有ROOT管理员权限，可通过
sudo -i命令切换； - 系统：适配CentOS/RHEL、Ubuntu/Debian系列Linux发行版，支持yum/apt包管理器；
- 基础环境：系统已预装Python3，离线无网络环境需手动提前安装gcc、make、dkms核心依赖；
- 硬件：适配昇腾Ascend310P、Ascend910A、Ascend910B系列NPU芯片。
快速使用步骤
限制要求
- 安装包的参数仅支持 --full、--install、--install-for-all
步骤1：为脚本添加可执行权限
进入脚本所在的技能根目录，执行以下命令为两个核心脚本赋予可执行权限：
chmod +x ./scripts/install_npu_driver.sh ./scripts/check_package.py
步骤2：执行自动化安装脚本
命令格式：./scripts/install_npu_driver.sh <NPU包文件夹完整路径> <驱动运行用户>
推荐示例（使用root用户进行安装）：
./scripts/install_npu_driver.sh /opt/ascend/npu_pkgs root
步骤3：系统重启（驱动生效必做）
安装完成后脚本会弹出交互式重启提示，NPU驱动和固件生效必须重启系统，无重启则无法完成驱动加载：
- 输入
y：系统立即重启，完成NPU驱动内核加载； - 输入
n：跳过立即重启，后续需手动执行reboot命令完成系统重启。
步骤 4：验证安装结果
系统重启后，执行昇腾官方原生命令验证 NPU 驱动加载状态：
npu-smi info
安装成功标识：命令输出内容包含NPU 芯片型号、Driver Version（驱动版本）、Firmware Version（固件版本），无任何报错信息。
核心脚本说明
check_package.py（Python 包校验脚本） 由 Shell 主脚本自动调用，无需手动执行，核心完成以下包校验工作： 检查安装包所在文件夹是否存在； 按昇腾官方正则匹配驱动 / 固件包，校验包数量唯一性； 验证包为有效文件（非目录）； 检测包的可执行权限，无权限时给出警告提示； 输出校验通过的驱动 / 固件包完整绝对路径，供 Shell 脚本调用。 install_npu_driver.sh（Shell 主安装脚本） 技能核心执行脚本，按固定流程自动化运行，全程无需人工干预，执行流程为：ROOT/Python3环境检查 → 包路径正则提取 → 包可执行权限按需赋权 → Python包二次校验 → 运行用户/组自动创建 → 系统依赖先验后装 → NPU驱动安装 → NPU固件安装 → 交互式重启确认 → npu-smi原生验证