MindSpeed LLM安装指导

本文主要向用户介绍如何快速基于MindSpore框架完成MindSpeed LLM（大语言模型分布式训练套件）的安装。

硬件配套和支持的操作系统

表 1 产品硬件支持列表

产品	是否支持
Atlas A3 训练系列产品	√
Atlas A3 推理系列产品	x
Atlas A2 训练系列产品	√
Atlas A2 推理系列产品	x
Atlas 200I/500 A2 推理产品	x
Atlas 推理系列产品	x
Atlas 训练系列产品	x

• 各硬件产品对应物理机部署场景支持的操作系统请参考兼容性查询助手。
• 各硬件产品对应虚拟机及容器部署场景支持的操作系统请参考《CANN 软件安装》的“操作系统兼容性说明”章节（商用版）或“操作系统兼容性说明”章节（社区版）。

安装前准备

请参见《版本说明》中的“相关产品版本配套说明”章节，下载安装对应的软件版本。

安装驱动固件

下载固件与驱动，请根据系统和硬件产品型号选择对应版本的社区版本或商用版本的固件与驱动。
参考如下命令安装：

chmod +x Ascend-hdk-<chip_type>-npu-driver_<version>_linux-<arch>.run
chmod +x Ascend-hdk-<chip_type>-npu-firmware_<version>.run
./Ascend-hdk-<chip_type>-npu-driver_<version>_linux-<arch>.run --full --force
./Ascend-hdk-<chip_type>-npu-firmware_<version>.run --full

安装CANN

请参考《CANN 快速安装》安装CANN软件（包含Toolkit、ops和NNAL包），并配置环境变量。

# 设置环境变量
source /usr/local/Ascend/cann/set_env.sh                 # 修改为实际安装的Toolkit包路径
source /usr/local/Ascend/nnal/atb/set_env.sh --cxx_abi=0 # 修改为实际安装的nnal包路径

[!NOTICE]
建议使用非root用户安装运行torch_npu，且建议对安装程序的目录文件做好权限管控：文件夹权限设置为750，文件权限设置为640。可以通过设置umask控制安装后文件的权限，如设置umask为0027。 更多安全相关内容请参见《安全声明》中各组件关于“文件权限控制”的说明。

安装MindSpore框架

参考MindSpore官方安装指导，根据系统类型、CANN版本及Python版本获取相应的安装命令以安装MindSpore 2.9.0，安装前请确保网络畅通。

安装MindSpeed LLM

请参考如下操作完成MindSpeed LLM及相关依赖的安装。

1. 使能环境变量。

   source /usr/local/Ascend/cann/set_env.sh                  # 修改为实际安装的Toolkit包路径
   source /usr/local/Ascend/nnal/atb/set_env.sh --cxx_abi=0  # 修改为实际安装的nnal包路径
   

2. 安装MindSpeed-Core-MS转换工具。

   git clone https://gitcode.com/ascend/MindSpeed-Core-MS.git -b master
   

3. 使用MindSpeed-Core-MS内部脚本提供配置环境。

   cd MindSpeed-Core-MS
   pip3 install -r requirements.txt  # 安装第三方依赖
   source auto_convert.sh llm        # 拉取训练所需组件库
   source tests/scripts/set_path.sh  # 设置环境变量