wangshuai09/ubuntu-cann7.0.0-torch21-py310-stable_diffusion 打包 Ubuntu、华为 CANN 7.0、Python 3.10 与 PyTorch 2.1，预置 Stable Diffusion 推理脚本与依赖。运行需匹配主机 CANN/驱动版本，通常 --device 映射 NPU 或指定 GPU 设备，内存与算力随模型分辨率显著增长。不适合无加速卡的常规 Kubernetes Deployment；边缘或单节点用对应 Device Plugin 或 docker runtime 部署，模型权重挂载只读 volume。

快速启动

docker run -d --name sd-cann \
  --device=/dev/davinci0 \
  -p 7860:7860 \
  -v $(pwd)/models:/models \
  docker.io/wangshuai09/ubuntu-cann7.0.0-torch21-py310-stable_diffusion:latest

推荐实践

docker run -d --name sd-cann \
  --restart unless-stopped \
  --device=/dev/davinci0 \
  -p 7860:7860 \
  -v /data/sd-models:/models:ro \
  -e MODEL_PATH=/models/sd-v1-5 \
  -e CANN_VISIBLE_DEVICES=0 \
  docker.io/wangshuai09/ubuntu-cann7.0.0-torch21-py310-stable_diffusion:v1.0

核心参数说明

• --device=/dev/davinci0 — 映射昇腾 NPU 设备
• -v .../models:/models:ro — SD 模型权重目录
• -e MODEL_PATH — 加载的 checkpoint 路径
• -e CANN_VISIBLE_DEVICES — 可见 NPU 编号
• -p 7860:7860 — Gradio/Web UI 端口
• 固定 tag — CANN 与主机驱动版本须一致