🐳 用Docker容器跑

创建于 2024-01-31 / 最近更新于 2024-01-31 / 5532
字体: [默认] [大] [更大] 收藏文档

Docker容器技术可以方便在多个平台部署Stable Diffusion WebUI。

将程序容器化的话,在不同Linux发行版跑Stable Diffusion WebUI就容易多了。

以Ubuntu 22.04为例,采用AbdBarho维护的docker-compose,仅支持Nvidia显卡。

  1. 安装Git、Docker,并确认已安装Nvidia驱动、CUDA、NVIDIA Container Toolkit套件
  2. 拷贝AbdBarho的保存库
git clone https://github.com/AbdBarho/stable-diffusion-webui-docker.git
cd stable-diffusion-webui-docker
  1. 安装依赖套件,过程中会自动下载一个Stable Diffusion的模型。
docker compose --profile download up --build
  1. 启动容器,选取auto代表启动AUTOMATIC1111开发的WebUI
docker compose --profile auto up --build
  1. 等待启动完成,用浏览器打开http://127.0.0.1:7860进入WebUI。要停止运行就是在终端机按Ctrl+C。

此docker-compose启动的Stable Diffusion WebUI,数据会挂载至同一目录下的data目录。

自订模型放到data/Stable-diffusion

生图的输出文件夹则是data/output

扩充功能请从网页界面装,或是在data新建extensions目录再于该处放入扩充功能的目录。

若要修改WebUI启动时的命令行参数,编辑此项目目录下的docker-compose.yml,修改CLI_ARGS这一行:

auto: &automatic
    <<: *base_service
    profiles: ["auto"]
    build: ./services/AUTOMATIC1111
    image: sd-auto:51
    environment:
      - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
35 人点赞过