软件安装

版本配套表

为了便于用户理解和选择合适的MindSpeed版本，我们提供了详细的版本配套表，如表1所示。

该表详细列出了MindSpeed版本与对应的PyTorch、Ascend Extension for PyTorch版本及CANN版本之间的匹配关系，确保用户能够根据自身软件环境准确选择相匹配的版本，以实现最优的性能与功能支持。

• 用户请根据需要选择MindSpeed代码分支下载源码并进行安装。
• 更多关于PyTorch和CANN的版本信息可点击昇腾辅助软件查看。

安装操作

1. 安装依赖的软件。

   表2 软件列表

   依赖软件	软件版本
   昇腾NPU驱动	建议下载并安装左侧软件，具体请参见《CANN 软件安装指南》。
   昇腾NPU固件
   Toolkit（开发套件）
   Kernels（算子包）
   NNAL（Ascend Transformer Boost加速库）
   PyTorch框架	建议下载并安装左侧软件，具体请参见《Ascend Extension for PyTorch 软件安装指南》。
   torch_npu插件
   apex

2. 下载MindSpeed源码2.0.0_core_r0.8.0分支。

   git clone -b 2.0.0_core_r0.8.0 https://gitee.com/ascend/MindSpeed.git

3. 安装MindSpeed。

   pip3 install -e MindSpeed

   

   如有旧版本MindSpeed，请先卸载，再进行安装操作。

4. 获取Megatron-LM并指定core_r0.8.0分支。

   git clone https://github.com/NVIDIA/Megatron-LM.git
   cd Megatron-LM
   git checkout core_r0.8.0

卸载

可参考如下命令：

pip3 uninstall -y mindspeed