国内Ubuntu环境Docker部署 ComfyUI

国内Ubuntu环境Docker部署 ComfyUI

---

趁着这两天用docker部署了 Stable Diffusion，顺手也安排上 ComfyUI。
ComfyUI相比 Stable Diffusion 原生的 WEB UI，更容易让人了解其出图的过程，极其适合学习与研究。拼接其强大的插件节点、不仅能够实现文生图、还可以配置图生图、图片生成视频等等，及其强大。

本笔记主要记录使用 docker 进行部署的相关文件，文件内容放在最后。

• Dockerfile
• compose.yaml
• requirements.txt
• start.sh

请在 ComfyUI 项目的根目录下创建一个docker文件夹，并将上述文件放入 docker 文件夹内。然后执行 cd docker && docker compose -f compose.yaml up。出现以下界面即部署成功。

如何使用 ComfyUI?

大家可以参考这个UP主的课程，讲得很清楚。
【全网最详细ComfyUI教程】！ComfyUI入门到精通全套教程 Stable Diffusion专业节点式界面新手教学教程(附安装包）

如何和 Stable Diffusion 共用模型？

大家只需要找到 ComfyUI 根目录中的 extra_model_paths_emample.yaml 文件，并在 a111 下配置 Stable Diffusion 的相关路径，最后重命名文件为 extra_model_paths_emample.yaml 即可。刷新页面就能在画面上看到 Stable Diffusion 项目中下载的模型。

Dockerfile

FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04

ENV LANG=C.UTF-8 LC_ALL=C.UTF-8

ENV DEBIAN_FRONTEN=noninteractive
SHELL ["/bin/bash", "-c"]

RUN apt-get update -y
RUN apt-get install -y libgl1-mesa-glx libglib2.0-0 gcc g++
RUN apt-get install -y net-tools wget curl git

RUN apt-get install -y make build-essential libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev libffi-dev liblzma-dev

# 从国内镜像源下载安装python
# wget https://www.python.org/ftp/python/3.10.13/Python-3.10.13.tar.xz && tar Jxf Python-3.10.13.tar.xz 
RUN wget https://mirrors.huaweicloud.com/python/3.10.13/Python-3.10.13.tar.xz && tar Jxf Python-3.10.13.tar.xz
RUN cd Python-3.10.13 && ./configure --with-system-ffi --enable-shared --enable-optimizations && make && make install && echo "/usr/local/lib" | tee /etc/ld.so.conf.d/python3.conf && ldconfig
RUN python3 -V && pip3 -V

# 设置国内镜像源
RUN pip3 config set global.index-url https://mirrors.aliyun.com/pypi/simple/ && pip3 config set install.trusted-host mirrors.aliyun.com

WORKDIR /workspace
COPY ./requirements.txt ./

RUN pip3 install -r requirements.txt
RUN pip3 install numpy==1.23.5

compose.yaml

services:
  comfyui:
    container_name: comfyui
    image: comfyui:1.0
    restart: always
    ports:
      - 8188:8188
    environment:
      - TZ=Asia/Tokyo
      - NVIDIA_VISIBLE_DEVICES=all
    volumes:
      - ../../ComfyUI:/workspace/ComfyUI
      - ../../automatic1111-stable-diffusion-webui:/workspace/automatic1111-stable-diffusion-webui
    # command: tail -f /dev/null
    command: sh -c "sh /workspace/ComfyUI/docker/start.sh"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              capabilities: [gpu]

requirements.txt

--extra-index-url https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/wheel/cu121/
# torch
torchsde
# torchvision
# torchaudio
torch==2.1.2
torchaudio==2.1.2
torchvision==0.16.2
einops
transformers>=4.28.1
tokenizers>=0.13.3
sentencepiece
safetensors>=0.4.2
aiohttp
pyyaml
Pillow
scipy
tqdm
psutil
#non essential dependencies:
kornia>=0.7.1
spandrel
soundfile

start.sh

#! /bin/bash
cd ComfyUI && python3 main.py --port 8188 --listen