Add Docker Instructions

This commit is contained in:
Mr. For Example
2024-07-12 15:09:01 -06:00
parent dacaf006d6
commit e06188372b
8 changed files with 283 additions and 124 deletions
+54 -8
View File
@@ -1,6 +1,6 @@
ARG CUDA_VERSION=12.1.0-devel
FROM docker.io/nvidia/cuda:${CUDA_VERSION}-ubuntu22.04
FROM --platform=amd64 docker.io/nvidia/cuda:${CUDA_VERSION}-ubuntu22.04
RUN apt-get update && \
apt-get install --no-install-recommends -y \
@@ -27,14 +27,18 @@ RUN apt-get update && \
ninja-build \
python3.11 \
python3.11-dev \
python3.11-venv \
wget \
&& \
curl -sS https://bootstrap.pypa.io/get-pip.py | python3.11 && \
rm -rf /var/lib/apt/lists/*
RUN ln -s /usr/bin/python3.11 /usr/bin/python & \
ln -s /usr/bin/python3.11 /usr/bin/python3 & \
ln -s /usr/bin/pip3.11 /usr/bin/pip
RUN python -m pip install --upgrade pip
RUN adduser --uid 1001 -q user && \
mkdir /app && chown user /app
USER user
@@ -53,35 +57,77 @@ ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,graphics
ENV PYOPENGL_PLATFORM egl
WORKDIR /app
# Clone ComfyUI repository
RUN git clone "https://github.com/comfyanonymous/ComfyUI.git" ./ && \
git reset --hard 05cd00695a84cebd5603a31f665eb7301fba2beb
git reset --hard 29c2e26724d4982a3e33114eb9064f1a11f4f4ed
# Setup a virtual environment
RUN python -m venv venv
# Activate the virtual environment
ENV PATH="/app/venv/bin:$PATH" \
VIRTUAL_ENV="/app/venv" \
PYTHONPATH="/app/venv/lib/python3.11/site-packages" \
PYTHONUSERBASE="/app/venv"
# Install the requirements
RUN pip install --no-cache -r requirements.txt
WORKDIR /app/custom_nodes/ComfyUI-3D-Pack/
COPY --chown=user:user requirements.txt ./
COPY --chown=user:user install.py ./
RUN pip install --no-cache -r requirements.txt \
# post requirements installation require gpu, setup
# `nvidia-container-runtime`, for docker, see
# https://stackoverflow.com/a/61737404
# those seem to be missed
ninja rembg[gpu] open_clip_torch \
python install.py
ninja rembg[gpu] open_clip_torch
COPY --chown=user:user ./ ./
RUN python install.py
# Install Custom Nodes
WORKDIR /app/custom_nodes/
# Essential nodes
RUN git clone "https://github.com/ltdrdata/ComfyUI-Impact-Pack" && \
cd ComfyUI-Impact-Pack && \
git reset --hard e1570e76799049fb0038f6232fd89800ab4565eb && \
git reset --hard ab17f8886945b0d36478950fe532164a8b569cc7 && \
pip install --no-cache -r requirements.txt
RUN git clone "https://github.com/kijai/ComfyUI-KJNodes" && \
cd ComfyUI-KJNodes && \
git reset --hard 95ae8b067a8aa5825471189954579ffa02aa9256 && \
git reset --hard ffafc9c2c675ce4e89386c725def331a88274004 && \
pip install --no-cache -r requirements.txt
RUN git clone "https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite" && \
cd ComfyUI-VideoHelperSuite && \
git reset --hard f94597739f20dac331f9918a245149b6b00a60f2 && \
git reset --hard 70faa9bcef65932ab72e7404d6373fb300013a2e && \
pip install --no-cache -r requirements.txt
COPY --chown=user:user ./ ComfyUI-3D-Pack/
# Extra nodes
RUN git clone "https://github.com/cubiq/ComfyUI_IPAdapter_plus" && \
cd ComfyUI_IPAdapter_plus && \
git reset --hard 13fedc634d1abf19d289fc7a7b5a74589465206c
RUN git clone "https://github.com/ssitu/ComfyUI_UltimateSDUpscale" --recursive && \
cd ComfyUI_UltimateSDUpscale && \
git reset --hard 70083f5d449c498ee0fb35f5293c91cebac4b758
RUN git clone "https://github.com/ltdrdata/ComfyUI-Inspire-Pack" && \
cd ComfyUI-Inspire-Pack && \
git reset --hard cadf604de528be62e4fbb1e3d12c51c98f20f50b && \
pip install --no-cache -r requirements.txt
RUN git clone "https://github.com/edenartlab/eden_comfy_pipelines" && \
cd eden_comfy_pipelines && \
git reset --hard 1b64dd507e8560466a8a50b8b8a704547890f525 && \
pip install --no-cache -r requirements.txt
RUN git clone "https://github.com/WASasquatch/was-node-suite-comfyui" && \
cd was-node-suite-comfyui && \
git reset --hard e036c1aa1b228c31473f78e020f47f0ce94d4c80 && \
pip install --no-cache -r requirements.txt
# Clone comfyui-manager to handle extra nodes
RUN git clone "https://github.com/ltdrdata/ComfyUI-Manager.git" && \
cd ComfyUI-Manager && \
git reset --hard 2b8e76197ae970dbd7854a09a5ef57731dc1c82f
WORKDIR /app
ENTRYPOINT [ "python", "main.py", "--listen", "0.0.0.0" ]