ascendai/mindiesd

MindIE-SD

快速参考

此镜像由https://www.hiascend.com/developer/software/mindie维护。

获取帮助：

• https://www.hiascend.com/developer/ascendhub/detail/af85b724a7e5469ebd7ea13c3439d48f
• MindIE-SD文档
• https://www.hiascend.com/developer
• 问题反馈

镜像概述

该镜像整合vLLM-Omni与**MindIE-SD（Mind推理引擎Stable Diffusion）**于单个容器中，支持在昇腾Atlas NPUs上进行多模态LLM推理和Stable Diffusion图像生成。

它基于quay.io/ascend/vllm-omni基础镜像构建（包含CANN 8.5.1、torch、torch_npu、vllm及vllm_ascend），提供两种针对不同NPU系列的变体：

• A2：基于quay.io/ascend/vllm-omni:v0.20.0，适用于昇腾Atlas A2系列NPU
• A3：基于quay.io/ascend/vllm-omni:v0.20.0-a3，适用于昇腾Atlas A3系列NPU

两种变体均添加以下昇腾调优与调试工具：

镜像标签与Dockerfile路径

标签命名规则

text
{版本}-cann{CAN版本}-{系列}-{操作系统}-py{Python版本}-{架构}

v3.0.0版本Dockerfile目录

每个系列均有专用Dockerfile，存放在MindIE-SD源码仓库的docker/omni目录中：

快速启动

拉取基础镜像

浏览所有可用标签：https://quay.io/repository/ascend/vllm-omni?tab=tags

A2系列：

bash
docker pull ***-quay.xuanyuan.run/ascend/vllm-omni:v0.20.0

A3系列：

bash
docker pull ***-quay.xuanyuan.run/ascend/vllm-omni:v0.20.0-a3

提示：若偏好使用Podman作为容器运行时，可将docker pull替换为podman pull。

运行容器

A2系列：

bash
docker run -it --rm --name=mindiesd \
    --privileged \
    --shm-size=1g \
    --device /dev/davinci0 \
    --device /dev/davinci1 \
    --device /dev/davinci2 \
    --device /dev/davinci3 \
    --device /dev/davinci_manager \
    --device /dev/devmm_svm \
    --device /dev/hisi_hdc \
    -v /usr/local/dcmi:/usr/local/dcmi \
    -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
    -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
    -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
    -v /etc/ascend_install.info:/etc/ascend_install.info \
    -v /root/.cache:/root/.cache \
    docker.xuanyuan.run/mindiesd:v3.0.0-A2-ubuntu22.04-py3.11-aarch64 \
    bash

A3系列：

bash
docker run -it --rm --name=mindiesd \
    --privileged \
    --shm-size=1g \
    --device /dev/davinci0 \
    --device /dev/davinci1 \
    --device /dev/davinci2 \
    --device /dev/davinci3 \
    --device /dev/davinci_manager \
    --device /dev/devmm_svm \
    --device /dev/hisi_hdc \
    -v /usr/local/dcmi:/usr/local/dcmi \
    -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
    -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
    -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
    -v /etc/ascend_install.info:/etc/ascend_install.info \
    -v /root/.cache:/root/.cache \
    docker.xuanyuan.run/mindiesd:v3.0.0-A3-ubuntu22.04-py3.11-aarch64 \
    bash

注意：--privileged标志和设备映射是NPU访问的必要条件。主机需将驱动库（/usr/local/Ascend/driver/lib64）、驱动版本信息、DCMI、npu-smi及昇腾安装信息挂载到容器中。

本地构建

克隆MindIE-SD仓库并从docker/omni目录构建：

A2系列：

bash
git clone https://gitcode.com/Ascend/MindIE-SD.git
cd MindIE-SD/docker/omni

docker build -t mindiesd:v3.0.0-A2-ubuntu22.04-py3.11-aarch64 \
    -f Dockerfile.a2.ubuntu .

A3系列：

bash
git clone https://gitcode.com/Ascend/MindIE-SD.git
cd MindIE-SD/docker/omni

docker build -t mindiesd:v3.0.0-A3-ubuntu22.04-py3.11-aarch64 \
    -f Dockerfile.a3.ubuntu .

自定义（二次开发）

若需添加自有依赖或应用代码，可基于此镜像创建新Dockerfile：

dockerfile
FROM docker.xuanyuan.run/mindiesd:v3.0.0-A2-ubuntu22.04-py3.11-aarch64

# 添加自定义包
RUN pip install --no-cache-dir your-package

# 复制应用代码
COPY ./your-app /workspace/your-app
WORKDIR /workspace/your-app

硬件支持

兼容性变更

请参考MindIE-SD文档获取最新发布说明及兼容性信息。

许可证与免责声明

此镜像基于**木兰宽松许可证2.0（Mulan PSL v2）**授权。完整文本请参见LICENSE文件。

通过拉取和使用此容器镜像，您同意华为容器许可协议的条款和条件。协议副本可在以下地址获取：https://www.hiascend.com/en/legal/ascendhub-download

您同意并承诺在使用此镜像中的华为或第三方软件时，将遵守相应华为或第三方软件的许可协议。