SuperAPI AI 工具集

cann-nnal-installer

部署与运维

查看项目

更新于 10-16 05:00

技能介绍

用途:在昇腾 NPU 环境中安装 CANN 完整组件(Toolkit、Kernels、NNAL),支持自定义安装路径,适用于 Docker 容器内搭建昇腾开发环境或安装指定版本(如 9.0.0-beta.2)。

安装顺序必须为 Toolkit → Kernels → NNAL,不可颠倒;NNAL 依赖 Toolkit 环境变量。

前置检查:先执行 npu-smi info 检查 NPU 驱动,无命令则需先装驱动与固件;据输出确认芯片型号以选择 Kernels 包(910B/910B3 用 910b、910 用 910、310P 用 310p、310B 用 310b);用 uname -m 确认架构(aarch64 选 _linux-aarch64.run,x86_64 选 _linux-x86_64.run)。

方式一(官网 run 包):从昇腾社区下载中心获取 toolkit、kernels、nnal 三个 run 包;依次 chmod +x 后执行 ./Ascend-cann-*.run --install --install-path="${INSTALL_PATH}",安装前先 source Toolkit 的 set_env.sh 再装 NNAL。官网下载需登录认证,无法直接 wget。

方式二(Docker 镜像提取):docker images | grep cann 确认镜像,无则 docker pull quay.io/ascend/cann:<版本>-<芯片>-<OS>-py<Python版本>;docker create 后 docker cp 提取 /usr/local/Ascend/cann-<版本> 与 /usr/local/Ascend/nnal,再 docker rm;需用 sed 将 set_env.sh 中硬编码的 /usr/local/Ascend/cann-<版本> 替换为实际路径。

环境变量:source cann 的 set_env.sh 与 nnal/atb/set_env.sh,可写入 ~/.bashrc 持久化;涉及 ASCEND_TOOLKIT_HOME、ASCEND_HOME_PATH、ASCEND_OPP_PATH、ATB_HOME_PATH、LD_LIBRARY_PATH、PYTHONPATH。

验证:检查 toolkit、ops、atb 版本信息,确认 bisheng、atc 可用,ASCEND_TOOLKIT_HOME 与 ATB_HOME_PATH 指向正确,npu-smi info 正常,最后输出安装验证报告。

限制与注意:三组件须装到同一路径且版本一致,CANN 版本须与宿主机 NPU 驱动兼容;常见问题包括目录权限不足、NNAL 找不到环境变量、镜像提取后命令找不到、Kernels 包与芯片型号不匹配。