ascendai/mindspeed-core

MindSpeed Core Docker镜像概述

快速参考

镜像标签关键字段

推荐标签模板：
{version}-{chip_info}-{os}-{python_tag}-{arch}

chip_info在镜像标签和CANN基础镜像名称中必须使用小写值，例如a3和910b。完整的--base-image值将直接传递，因此其标签必须与已发布的CANN镜像名称完全匹配。

示例：

• 0.12.1-910b-ubuntu22.04-py3.11-x86_64
• master-a3-openeuler24.03-py3.11-aarch64

Dockerfile归档路径

• docker/Dockerfile
• docker/build.sh

构建选项

推荐使用docker/build.sh作为构建入口，支持通过操作系统、NPU类型、Python标签和CANN基础镜像版本选择CANN基础镜像。

快速开始

按默认配置构建

bash
cd docker
bash build.sh

构建a3 + openEuler（基于CANN 9.0基础镜像）

bash
cd docker
bash build.sh -t a3 -o openeuler24.03 --base-image-version 9.0.0-beta.2

使用完整基础镜像名称构建

脚本会尽可能从标签中自动检测NPU类型、操作系统和Python版本：

bash
cd docker
bash build.sh \
  --base-image swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:9.0.0-beta.2-910b-openeuler24.03-py3.11

运行示例

使用docker images中的REPOSITORY:TAG值，例如mindspeed-core:master-910b-openeuler24.03-py3.11-aarch64：

bash
docker run -itd \
  --name mindspeed \
  --privileged \
  --network host \
  --ipc=host \
  -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
  -v /usr/local/dcmi:/usr/local/dcmi \
  -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
  -v /etc/ascend_install.info:/etc/ascend_install.info \
  -v /home:/home \
  -v /data:/data \
  -v /mnt:/mnt \
  docker.xuanyuan.run/mindspeed-core:master-910b-openeuler24.03-py3.11-aarch64

进入运行中的容器

bash
docker exec -it mindspeed /bin/bash

如果主机上的npu-smi安装在/usr/local/sbin/npu-smi路径下，请相应替换挂载路径。

兼容性说明

• 该镜像使用统一的Dockerfile和构建脚本，支持可配置的CANN基础镜像选择。
• 默认基础镜像使用CANN 9.0.0-beta.2、910b、openEuler 24.03和Python 3.11。
• 可通过docker/build.sh切换到Ubuntu 22.04、a3或其他CANN基础镜像版本。
• MindSpeed克隆到/MindSpeed；Megatron-LM克隆到/Megatron-LM。
• 安装了PyTorch、torch_npu、MindSpeed Core、Megatron-LM及requirements.txt中的Python依赖。

许可协议

MindSpeed以Apache License 2.0协议发布，详情见LICENSE文件。

与所有Docker镜像一样，这些镜像可能还包含其他协议下的软件，例如基础发行版中的Bash以及包含的主软件的任何直接或间接依赖项。

对于预构建镜像的任何使用，镜像用户有责任确保该镜像的使用符合其中包含的所有软件的相关许可协议。