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 天前
SD的细分功能包括重绘,图像处理、放大等扩散模型应用
人工智能·python·深度学习·机器学习·stable diffusion·comfyui·ai工作流
二院大蛙5 天前
Stable Diffusion 3.5 FP8在农业无人机航拍模拟图中的地形还原精度
stable diffusion· fp8· 农业无人机
或困5 天前
Stable Diffusion 3.5 FP8镜像支持灰度检测与异常报警
stable diffusion·灰度发布·fp8量化
沉默的大羚羊5 天前
Stable Diffusion 3.5 FP8模型可用于旅游宣传海报制作
stable diffusion·文生图·fp8
BOBO爱吃菠萝5 天前
Stable Diffusion 3.5 FP8镜像自动化部署脚本发布
stable diffusion·量化·fp8
九章云极AladdinEdu5 天前
项目分享|SD-Trainer:Stable Diffusion 训练集成工具
stable diffusion·端到端学习·高斯泼溅·3d场景分割·物体级代码本·2d到3d提升
qq_420443275 天前
AMD显卡在windows中通过WSL安装使用stable diffusion(WebUI和ComfyUI)
linux·windows·ubuntu·stable diffusion·wsl
网安入门学习5 天前
2025年AIGC人才需求报告:从招聘数据看行业趋势与技能要求
人工智能·windows·ai作画·stable diffusion·aigc
ai_xiaogui5 天前
Stable Diffusion Web UI 整合包一键安装教程:Windows/Mac零基础部署AI绘画工具
人工智能·ai作画·stable diffusion·一键整合包·ai生图神器·ai生图和动作迁移
人工智能AI技术6 天前
【SD教程】提示词
人工智能·stable diffusion·aigc·ai绘画