MindSpeed-MM MindSpore后端安装指导

版本配套

软件	版本	安装指南
Python	>= 3.10
Driver	AscendHDK 25.5.0	《驱动固件安装指南 》
Firmware	AscendHDK 25.5.0
CANN	CANN 8.5.0	《CANN 软件安装指南 》
MindSpore	2.7.2	《MindSpore安装》

驱动固件安装

下载驱动固件，请根据系统和硬件产品型号选择对应版本的 driver和 firmware。参考安装NPU驱动固件或执行以下命令安装：

bash Ascend-hdk-*-npu-firmware_*.run --full
bash Ascend-hdk-*-npu-driver_*.run --full --force

CANN安装

下载CANN，请根据系统选择 aarch64或 x86_64对应版本的 cann-toolkit、cann-kernel和 cann-nnal。参考CANN安装或执行以下命令安装：

bash Ascend-cann-toolkit_8.5.0_linux-aarch64.run --install
bash Ascend-cann-*-ops_8.5.0_linux-aarch64.run --install
source /usr/local/Ascend/cann/set_env.sh # 安装nnal包需要source环境变量
bash Ascend-cann-nnal_8.5.0_linux-aarch64.run --install
# 设置环境变量
source /usr/local/Ascend/nnal/asdsip/set_env.sh
source /usr/local/Ascend/nnal/atb/set_env.sh --cxx_abi=0

MindSpore安装

参考MindSpore官方安装指导，根据系统类型、CANN版本及Python版本选择相应的安装命令进行安装，安装前请确保网络畅通。或执行以下命令安装：

pip install mindspore==2.7.2

代码一键适配

MindSpeed-Core-MS提供了代码、环境的一键适配功能，执行以下命令完成一键适配后，用户即可开启基于MindSpore AI套件的多模态模型之旅。

git clone https://gitcode.com/Ascend/MindSpeed-Core-MS.git -b r0.5.0
cd MindSpeed-Core-MS
source auto_convert.sh mm
cd MindSpeed-MM