SuperAPI AI 工具集

external-gitcode-ascend-ascend-npu-driver-install

编程与开发

查看项目

更新于 10-15 17:19

技能介绍

用途:端到端自动化完成昇腾NPU驱动与固件安装部署,覆盖安装包校验到驱动生效验证全流程,遵循昇腾官方安装规范,适配主流Linux发行版。

主要能力:按官方正则从指定文件夹提取驱动/固件包并强制校验数量唯一(仅1个驱动包+1个固件包);安装包可执行权限按需赋权,无权限时自动chmod +x并二次校验;Python+Shell双重包校验,提前验证包格式、路径与文件有效性;系统依赖先验后装;严格按先驱动后固件顺序安装;安装后提供交互式重启选项,并用官方npu-smi命令验证驱动状态。

使用方式:将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权限(sudo -i切换),系统预装Python3,离线环境需提前安装gcc、make、dkms。执行:chmod +x ./scripts/install_npu_driver.sh ./scripts/check_package.py,然后./scripts/install_npu_driver.sh /opt/ascend/npu_pkgs root。安装完成后按提示重启(输入y立即重启,输入n需手动reboot),重启后执行npu-smi info验证,输出含芯片型号、Driver Version、Firmware Version且无报错即成功。

适用场景:CentOS/RHEL、Ubuntu/Debian系列Linux发行版,支持yum/apt;适配Ascend310P、Ascend910A、Ascend910B系列NPU芯片。

限制:安装包参数仅支持--full、--install、--install-for-all;必须ROOT权限;驱动与固件生效必须重启系统;该版本无需校验系统内核;安装包文件夹不得含无关文件。其余细节暂未提供。