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 .

相关推荐
迈火21 小时前
ComfyUI-3D-Pack:3D创作的AI神器
人工智能·gpt·3d·ai·stable diffusion·aigc·midjourney
Seeklike2 天前
diffusers学习--stable diffusion的管线解析
人工智能·stable diffusion·diffusers
马甲是掉不了一点的<.<2 天前
Stable Diffusion 环境配置详细指南
stable diffusion·环境配置
软件测试-阿涛2 天前
【AI绘画】Stable Diffusion webUI 常用功能使用技巧
人工智能·深度学习·计算机视觉·ai作画·stable diffusion
m0_603888713 天前
Stable Diffusion Models are Secretly Good at Visual In-Context Learning
人工智能·ai·stable diffusion·论文速览
爱分享的飘哥15 天前
第三十七章:文生图的炼金术:Stable Diffusion完整工作流深度解析
人工智能·pytorch·stable diffusion·文生图·ai绘画·代码实战·cfg
EndingCoder18 天前
Three.js + AI:结合 Stable Diffusion 生成纹理贴图
开发语言·前端·javascript·人工智能·stable diffusion·ecmascript·three.js
那年一路北18 天前
Deforum Stable Diffusion,轻松实现AI视频生成自由!
人工智能·stable diffusion·音视频
全宝18 天前
🎨【AI绘画实战】从零搭建Stable Diffusion环境,手把手教你生成超可爱Q版大头照!
人工智能·python·stable diffusion
sculida24 天前
秋叶sd-webui频繁出现生成后无反应的问题
stable diffusion