ARM架构推理Stable Diffusiond

代码仓库:

https://github.com/siutin/stable-diffusion-webui-docker.git

Docker容器地址:

https://hub.docker.com/r/siutin/stable-diffusion-webui-docker/tags

git clone https://github.com/siutin/stable-diffusion-webui-docker.git

cd stable-diffusion-webui-docker

准备主机中的目录映射:

创建目录

mkdir -p /MY-DATA-DIR && cd /MY-DATA-DIR

mkdir models outputs

sudo chown 10000:$UID -R models outputs

sudo chmod 775 -R models outputs

使用 CUDA 运行

With the latest CUDA version

docker run -it --name sdw --gpus all --network host \

-v $(pwd)/models:/app/stable-diffusion-webui/models \

-v $(pwd)/outputs:/app/stable-diffusion-webui/outputs \

--rm siutin/stable-diffusion-webui-docker:latest-cuda \

bash webui.sh --share

With specific CUDA version

docker run -it --name sdw --gpus all --network host \

-v $(pwd)/models:/app/stable-diffusion-webui/models \

-v $(pwd)/outputs:/app/stable-diffusion-webui/outputs \

--rm siutin/stable-diffusion-webui-docker:latest-cuda-12.6.2 \

bash webui.sh --share

仅使用 CPU 运行

docker run -it --name sdw --network host \

-v $(pwd)/models:/app/stable-diffusion-webui/models \

-v $(pwd)/outputs:/app/stable-diffusion-webui/outputs \

--rm siutin/stable-diffusion-webui-docker:latest-cpu \

bash webui.sh --skip-torch-cuda-test --use-cpu all --share

构建自定义镜像

Nvidia CUDA image

nvidia-docker buildx build -f Dockerfile.cuda \

--platform linux/amd64 \

--build-arg BUILD_DATE=$(date -u +'%Y-%m-%dT%H:%M:%SZ') \

--build-arg BUILD_VERSION=custom-cuda \

-t siutin/stable-diffusion-webui-docker:custom-cuda .

Nvidia CUDA image with specific version

nvidia-docker buildx build -f Dockerfile.cuda \

--platform linux/amd64 \

--build-arg BUILD_DATE=$(date -u +'%Y-%m-%dT%H:%M:%SZ') \

--build-arg BUILD_VERSION=custom-cuda \

--build-arg CUDA_VERSION=12.5.1 \

-t siutin/stable-diffusion-webui-docker:custom-cuda-12.5 .

CPU only image

docker buildx build -f Dockerfile.cpu \

--platform linux/arm64 \

--build-arg BUILD_DATE=$(date -u +'%Y-%m-%dT%H:%M:%SZ') \

--build-arg BUILD_VERSION=custom-cpu \

-t siutin/stable-diffusion-webui-docker:custom-cpu .

docker run -it --name sdw --gpus all --network host \
-v /home/ai/Project/SD模型/sd-webui
/models:/app/stable-diffusion-webui/models \
-v /home/ai/Project/SD模型/sd-webui
/outputs:/app/stable-diffusion-webui/outputs \
--rm siutin/stable-diffusion-webui-docker :cpu-v1.10.1-2025-02-10 \
bash webui.sh --share

Ubuntu 20.04 focal 版本(适合 ARM 64)

sudo tee /etc/apt/sources.list > /dev/null << EOF

默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ focal main restricted universe multiverse

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ focal main restricted universe multiverse

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ focal-updates main restricted universe multiverse

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ focal-updates main restricted universe multiverse

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ focal-backports main restricted universe multiverse

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ focal-backports main restricted universe multiverse

安全更新源

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ focal-security main restricted universe multiverse

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ focal-security main restricted universe multiverse

Docker 镜像源(arm64)

deb [arch=arm64] https://mirrors.aliyun.com/docker-ce/linux/ubuntu focal stable

deb-src [arch=arm64] https://mirrors.aliyun.com/docker-ce/linux/ubuntu focal stable

EOF

Ubuntu 22.04 jammy 版本(适合 ARM 64)

sudo tee /etc/apt/sources.list > /dev/null << EOF

默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy main restricted universe multiverse

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy main restricted universe multiverse

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy-updates main restricted universe multiverse

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy-updates main restricted universe multiverse

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy-backports main restricted universe multiverse

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy-backports main restricted universe multiverse

安全更新源

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy-security main restricted universe multiverse

deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ jammy-security main restricted universe multiverse

Docker 镜像源(arm64)

deb [arch=arm64] https://mirrors.aliyun.com/docker-ce/linux/ubuntu jammy stable

deb-src [arch=arm64] https://mirrors.aliyun.com/docker-ce/linux/ubuntu jammy stable

EOF

构建镜像

docker buildx build -f Dockerfile.cuda --platform linux/arm64 --build-arg BUILD_DATE=$(date -u +'%Y-%m-%dT%H:%M:%SZ') --build-arg BUILD_VERSION=custom-cuda-12.4 --build-arg CUDA_VERSION=12.4.0 -t siutin/stable-diffusion-webui-docker:custom-cuda-12.4 .

相关推荐
心疼你的一切3 小时前
基于CANN仓库算力手把手实现Stable Diffusion图像生成(附完整代码+流程图)
数据仓库·深度学习·stable diffusion·aigc·流程图·cann
Niuguangshuo1 天前
DALL-E 3:如何通过重构“文本描述“革新图像生成
人工智能·深度学习·计算机视觉·stable diffusion·重构·transformer
Niuguangshuo2 天前
深入解析 Stable Diffusion XL(SDXL):改进潜在扩散模型,高分辨率合成突破
stable diffusion
Niuguangshuo2 天前
深入解析Stable Diffusion基石——潜在扩散模型(LDMs)
人工智能·计算机视觉·stable diffusion
迈火2 天前
SD - Latent - Interposer:解锁Stable Diffusion潜在空间的创意工具
人工智能·gpt·计算机视觉·stable diffusion·aigc·语音识别·midjourney
迈火9 天前
Facerestore CF (Code Former):ComfyUI人脸修复的卓越解决方案
人工智能·gpt·计算机视觉·stable diffusion·aigc·语音识别·midjourney
重启编程之路10 天前
Stable Diffusion 参数记录
stable diffusion
孤狼warrior13 天前
图像生成 Stable Diffusion模型架构介绍及使用代码 附数据集批量获取
人工智能·python·深度学习·stable diffusion·cnn·transformer·stablediffusion
love530love15 天前
【避坑指南】提示词“闹鬼”?Stable Diffusion 自动注入神秘词汇 xiao yi xian 排查全记录
人工智能·windows·stable diffusion·model keyword
世界尽头与你15 天前
Stable Diffusion web UI 未授权访问漏洞
安全·网络安全·stable diffusion·渗透测试