用途:端到端自动化完成昇腾NPU驱动与固件安装,覆盖从安装包校验到驱动生效验证的全流程,遵循昇腾官方安装规范,适配主流Linux发行版。
主要能力:基于官方正则自动提取指定文件夹内的驱动/固件包,强制校验仅1个驱动包+1个固件包;安装包可执行权限按需赋权,无权限时自动chmod +x并二次校验;Python+Shell双重包校验,提前验证包格式、路径、文件有效性;系统依赖先验后装;严格按先驱动后固件顺序安装,安装后提供交互式重启选项,并用官方npu-smi验证驱动状态。适配Ascend310P、Ascend910A、Ascend910B系列芯片。
前置准备:脚本check_package.py与install_npu_driver.sh放在同一目录,根目录建议命名ascend-npu-driver-install。包文件夹内仅放1个符合官方命名格式的驱动.run包和1个固件.run包,格式为Ascend-hdk-<chip_type>-npu-driver_<version>linux-<arch>.run和Ascend-hdk-<chip_type>-npu-firmware<version>.run。需ROOT权限,已装Python3;离线环境需提前安装gcc、make、dkms。
使用步骤:第一步执行chmod +x ./scripts/install_npu_driver.sh ./scripts/check_package.py;第二步执行./scripts/install_npu_driver.sh <NPU包文件夹完整路径> <驱动运行用户>,示例./scripts/install_npu_driver.sh /opt/ascend/npu_pkgs root;第三步安装完成后按提示输入y立即重启或输入n后手动reboot,重启为驱动生效必做;第四步重启后执行npu-smi info,输出含芯片型号、Driver Version、Firmware Version且无报错即成功。
限制:安装包参数仅支持--full、--install、--install-for-all;该版本无需校验系统内核;必须重启系统才能完成驱动加载。其余功能、收费与平台信息暂未提供。