ComfyUI容器化部署

2025-04-04
ComfyUI

在容器中访问宿主机NVIDIA GPU显卡

在容器中访问显卡需要NVIDIA Container Runtime

1
curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | sudo apt-key add -
2
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
3
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
4

5
sudo apt-get update
6
sudo apt-get install nvidia-container-runtime

修改docker配置文件 vim /etc/docker/daemon.json

1
{
2
"default-runtime": "nvidia",
3
    "runtimes": {
4
        "nvidia": {
5
            "path": "nvidia-container-runtime",
6
            "runtimeArgs": []
7
        }
8
    },
9
}

然后重启Docker，启动容器时携带标志 --runtime=nvidia 即可使用宿主机GPU

编写Dockerfile

1
FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime
2
LABEL authors="jianzhang"
3

4
#更换为国内软件源
5
COPY source.list /etc/apt/sources.list
6
#安装需要的软件包
7
RUN apt update && apt install -y software-properties-common && add-apt-repository ppa:deadsnakes/ppa && apt update && DEBIAN_FRONTEND=noninteractive apt install -y build-essential libcurl4-openssl-dev gnutls-bin git-all wget curl
8
#设置git
9
RUN git config --global http.version HTTP/1.1;git config --global http.sslVerify false;git config --global http.postBuffer 1048576000
10
#下载ComfyUI并安装依赖
11
#RUN cd /workspace && git clone git://github.com/comfyanonymous/ComfyUI.git
12
COPY ComfyUI /workspace/ComfyUI
13
RUN cd /workspace/ComfyUI && pip install xformers insightface diffusers ultralytics -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --extra-index-url https://download.pytorch.org/whl/cu124 --extra-index-url https://pypi.python.org/simple --trusted-host mirrors.aliyun.com
14
#下载ComfyUI-Manager并安装依赖
15
#RUN cd /workspace/ComfyUI/custom_nodes && git clone git://github.com/ltdrdata/ComfyUI-Manager.git
16
COPY ComfyUI-Manager /workspace/ComfyUI/custom_nodes/ComfyUI-Manager
17
RUN cd /workspace/ComfyUI/custom_nodes/ComfyUI-Manager && pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --extra-index-url https://pypi.python.org/simple --trusted-host mirrors.aliyun.com
18

19
VOLUME /workspace/ComfyUI/models/
20
VOLUME /workspace/ComfyUI/custom_nodes/
21

22
EXPOSE 8188
23
ENTRYPOINT ["python","/workspace/ComfyUI/main.py"]
24
HEALTHCHECK --interval=8s --timeout=5s CMD curl -fs http://localhost:8188/ || exit 1

构建镜像

1
docker buildx build --no-cache  -t jianzhang/comfyui:2.6.0-cuda12.4-cudnn9-runtime ./

运行容器

1
docker run --runtime=nvidia -p 8188:8188  -d jianzhang/comfyui:2.6.0-cuda12.4-cudnn9-runtime