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 .

相关推荐
ding_zhikai12 小时前
stable-diffusion试验1-静态人物
stable diffusion
love530love3 天前
【笔记】 Podman Desktop 中部署 Stable Diffusion WebUI (GPU 支持)
人工智能·windows·笔记·python·容器·stable diffusion·podman
迈火4 天前
PuLID_ComfyUI:ComfyUI中的图像生成强化插件
开发语言·人工智能·python·深度学习·计算机视觉·stable diffusion·语音识别
ding_zhikai5 天前
SD:在一个 Ubuntu 系统安装 stable diffusion Web UI
前端·ubuntu·stable diffusion
不当菜鸡的程序媛5 天前
Stable Diffusion里面Cross-Attention设计:为啥Q来自图像/噪声,K和V来自文本
人工智能·计算机视觉·stable diffusion
云澈ovo9 天前
FP16混合精度训练:Stable Diffusion生成速度提升300%的硬件配置方案
人工智能·机器学习·stable diffusion
weiwei228441 个月前
文生图模型Stable Diffusion使用详解
stable diffusion·文生图
Hello123网站1 个月前
DreamStudio-Stable Diffusion官方绘画工具
stable diffusion·ai工具
如若1231 个月前
CompVis Stable Diffusion 正确安装与运行指南
stable diffusion
全息数据1 个月前
DDPM代码讲解【详细!!!】
深度学习·stable diffusion·多模态·ddpm