前言

comfyui官方没提供docker镜像…而其他docker镜像又不太确定是否符合自己的配置，于是写篇文档记录下如何打包

基础镜像

首先要有个ubuntu基础镜像，不然连python都没有的话没法运行
构建ubuntu基础镜像见这篇：Docker打包本地服务

安装

我创建了两个独立的文件，一个是venv，另一个是comfyui的源码解压

# 创建虚拟环境
python -m venv comfyui-venv
cd comfyui-venv
# 需要和nvidia-smi显示的cuda版本一致
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu13

# 安装依赖
cd ../ComfyUI-0.3.76
pip install -r requirements.txt

pip下载问题见这篇： wget pip git git-lfs下载失败报错解决

打包docker

将comfyui的源码路径和venv拷贝到构建Docker的路径下，名为comfyui的文件夹里

启动脚本

source /home/arco/comfyui/comfyui-0.3.76-venv/bin/activate
python /home/arco/comfyui/ComfyUI-0.3.76/main.py --listen
# 如果只有cpu的话用 python /home/arco/comfyui/ComfyUI-0.3.27/main.py --listen --cpu

Docker file

FROM ubuntu22.04-base

# 拷贝时指定所有权
COPY --chown=arco:arco comfyui /home/arco/comfyui/
COPY start.sh /

WORKDIR /
CMD ["/bin/bash", "/start.sh"]

docker build -t comfyui-0376 .

启动

export docker_image="comfyui-0376"
docker run -d \
--cpus=4 \
--gpus device=1 \
-p 8188:8188 \
-v /data2/models/comfyui:/home/arco/comfyui/ComfyUI-0.3.76/models \
$docker_image \

（使用-v指定存放模型的目录（和comfyui源码路径中的models同结构））