用途:在昇腾 NPU 环境(含 Docker 容器)中安装 NNAL(ATB 加速库),并完成环境变量配置与安装验证;也可用于安装或升级指定版本的 NNAL。
前置条件:必须已通过 cann-operator-env-config 技能完成 Toolkit 与 Kernels 安装,确保 /usr/local/Ascend/ascend-toolkit/set_env.sh 存在。另需从用户获取 NNAL 安装目标路径 INSTALL_PATH(如 /usr/local/Ascend);未提供时必须停止并要求用户提供。
主要步骤:
- 检查 NPU 驱动:npu-smi info,不可用或未检测到设备则报错停止。
- 若 Toolkit+Kernels 未安装,先调用 cann-operator-env-config(支持离线 run 包、conda 在线、yum 在线三种方式)。
- source /usr/local/Ascend/ascend-toolkit/set_env.sh,并可用 echo $ASCEND_TOOLKIT_HOME 验证;NNAL 安装脚本依赖该环境变量。
- 安装 NNAL:run 包方式为 chmod +x Ascend-cann-nnal_.run 后执行 ./Ascend-cann-nnal_.run --install --install-path="${INSTALL_PATH}";部分包询问确认时可用 yes | 前置实现非交互安装。也可从 Docker 镜像提取:docker create 后 docker cp cann_nnal_temp:/usr/local/Ascend/nnal "${INSTALL_PATH}/",再 docker rm。
- 加载环境变量:source "${INSTALL_PATH}/nnal/atb/set_env.sh",可追加到 ~/.bashrc 持久化。
- 验证:检查 nnal/atb 目录、Toolkit 版本文件、NNAL 版本文件,确认 bisheng、atc 可用,并再次执行 npu-smi info。
环境变量:ASCEND_TOOLKIT_HOME、ATB_HOME_PATH、LD_LIBRARY_PATH 应正确设置。参考耗时与体积:NNAL run 包约 500-600MB,安装约 5-10 分钟,含 Toolkit+Kernels 总计约 15-20 分钟。
限制与常见问题:需 root 权限时使用 sudo;Docker 中提示 driver package is not installed 属正常(使用宿主机驱动);npu-smi 异常多因驱动未装或容器未挂载 NPU 设备(如 --device=/dev/davinciX);ASCEND_TOOLKIT_HOME 未设置多因 Toolkit 未装或 set_env.sh 路径不对。
官方资料暂未提供收费、许可与具体版本支持范围信息。