Atb Nnal Installer

CANN NNAL 安装部署

功能概述

在昇腾 NPU 环境中安装 NNAL（ATB 加速库）。本技能假设 Toolkit+Kernels 已由 cann-operator-env-config 技能安装完毕。

调用时机

• 用户已完成 Toolkit+Kernels 安装（通过 cann-operator-env-config），需要安装 NNAL
• 用户在 Docker 容器中需要安装 NNAL
• 用户需要安装或升级特定版本的 NNAL

前置条件

必须已执行 cann-operator-env-config 技能，确保：

• Toolkit 已安装，/usr/local/Ascend/ascend-toolkit/set_env.sh 存在
• Kernels（ops）已安装

⚠️ 路径约束（必须执行）

执行此技能前，必须从用户处获取以下路径：

• <INSTALL_PATH>: NNAL 安装目标路径（如 /usr/local/Ascend）

若用户未提供 INSTALL_PATH，立即停止并打断，要求用户提供安装路径。

脚本校验示例：

bashCopy code
if [ -z "$INSTALL_PATH" ]; then
    echo "ERROR: INSTALL_PATH 未设置，请提供 NNAL 安装路径。"
    exit 1
fi

详细内容

第1步：检查 NPU 驱动

bashCopy code
npu-smi info

若 npu-smi 不可用或未检测到 NPU 设备，停止安装并报错。

第2步：调用 cann-operator-env-config（如尚未安装 Toolkit+Kernels）

如果 Toolkit+Kernels 尚未安装，首先调用 cann-operator-env-config 技能安装 Toolkit 和 Kernels：

• 该技能位于 /home/s30073260/swiglu_quant/agent-skills-ivan-fork/skills/cann-operator-env-config/
• 提供 3 种安装方式：离线 run 包、conda 在线、yum 在线
• 安装完成后，Toolkit 和 Kernels 将位于 /usr/local/Ascend/

第3步：Source Toolkit 环境变量

bashCopy code
source /usr/local/Ascend/ascend-toolkit/set_env.sh

关键：NNAL 安装脚本依赖 ASCEND_TOOLKIT_HOME 等环境变量，必须先 source。

验证环境变量：

bashCopy code
echo $ASCEND_TOOLKIT_HOME
# 应输出: /usr/local/Ascend/ascend-toolkit/latest 或自定义路径

第4步：安装 NNAL

注意：以下命令使用用户提供的 <INSTALL_PATH>，如未提供则已在路径约束步骤中打断。

方式一：run 包安装（推荐）

bashCopy code
chmod +x Ascend-cann-nnal_*.run
./Ascend-cann-nnal_*.run --install --install-path="${INSTALL_PATH}"

方式二：从 Docker 镜像提取

bashCopy code
CANN_IMAGE="quay.io/ascend/cann:<版本>-<芯片>-<OS>-py<Python版本>"

# 示例
# CANN_IMAGE="quay.io/ascend/cann:9.0.0-beta.2-910-openeuler24.03-py3.11"

docker create --name cann_nnal_temp "${CANN_IMAGE}"
docker cp cann_nnal_temp:/usr/local/Ascend/nnal "${INSTALL_PATH}/"
docker rm cann_nnal_temp

第5步：加载环境变量

bashCopy code
source "${INSTALL_PATH}/nnal/atb/set_env.sh"

持久化到 ~/.bashrc：

bashCopy code
cat >> ~/.bashrc << EOF
source ${INSTALL_PATH}/nnal/atb/set_env.sh
EOF

第6步：安装验证

bashCopy code
# 检查 NNAL 目录结构
ls -lh ${INSTALL_PATH}/nnal/atb/

# 验证 Toolkit 版本
cat /usr/local/Ascend/ascend-toolkit/latest/version.cfg 2>/dev/null || \
    cat /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/ascend_toolkit_install.info

# 验证 NNAL 版本
cat ${INSTALL_PATH}/nnal/atb/version 2>/dev/null || echo "检查 NNAL 安装目录"

# 验证关键命令
which bisheng && bisheng --version
which atc && atc --version

# 验证 NPU 状态
npu-smi info

环境变量汇总

安装完成后，以下环境变量应正确设置：

参考信息

部分 run 包会询问确认，使用 yes | 管道实现非交互安装：

bashCopy code
yes | ./Ascend-cann-nnal_*.run --install --install-path="${INSTALL_PATH}"

安装时若出现 "driver package is not installed" 警告，属于正常现象——Docker 容器使用宿主机驱动，无需安装驱动包。

常见问题

原因：NPU 驱动未安装或 Docker 容器未挂载 NPU 设备。

解决：

1. 确认宿主机 NPU 驱动正常：宿主机执行 npu-smi info
2. Docker 容器启动时添加 --device=/dev/davinciX 参数

原因：Toolkit 未安装或 set_env.sh 路径不正确。

解决：

1. 确认 /usr/local/Ascend/ascend-toolkit/latest/set_env.sh 存在
2. 如不存在，返回执行 cann-operator-env-config 安装 Toolkit+Kernels

原因：安装路径需 root 权限。

解决：

bashCopy code
# 使用 root 权限安装
sudo ./Ascend-cann-nnal_*.run --install --install-path=/usr/local/Ascend

执行结果

⚠️ 执行后填写：技能执行完成后，参照下方格式填写实际执行结果。

检查点检查表

VERDICT: ✅ SUCCESS / ⚠️ PARTIAL / ❌ FAILED / ⏭️ SKIPPED

问题列表（若有）

执行摘要

• 执行时间：
• 执行环境：
• NNAL 版本：
• 持续时间：
• 通过率：