88 lines
2.6 KiB
Docker
88 lines
2.6 KiB
Docker
ARG CUDA_VERSION=12.1.0-devel
|
|
|
|
FROM docker.io/nvidia/cuda:${CUDA_VERSION}-ubuntu22.04
|
|
|
|
RUN apt-get update && \
|
|
apt-get install --no-install-recommends -y \
|
|
build-essential \
|
|
curl \
|
|
ffmpeg \
|
|
git \
|
|
# TODO not sure all of this is required, remove unnnecessary
|
|
libegl1 \
|
|
libegl1-mesa-dev \
|
|
libgl1 \
|
|
libglib2.0-0 \
|
|
libgl1-mesa-dev \
|
|
libgl1-mesa-glx \
|
|
libgles2 \
|
|
libgles2-mesa-dev \
|
|
libglib2.0-0 \
|
|
libglvnd-dev \
|
|
libglvnd0 \
|
|
libglx0 \
|
|
libsm6 \
|
|
libxext6 \
|
|
libxrender1 \
|
|
ninja-build \
|
|
python3.11 \
|
|
python3.11-dev \
|
|
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/pip3.11 /usr/bin/pip
|
|
|
|
RUN adduser --uid 1001 -q user && \
|
|
mkdir /app && chown user /app
|
|
USER user
|
|
|
|
ENV PYTHONDONTWRITEBYTECODE=1
|
|
ENV PYTHONUNBUFFERED=1
|
|
|
|
# for GLEW
|
|
ENV LD_LIBRARY_PATH /usr/lib64:$LD_LIBRARY_PATH
|
|
|
|
# nvidia-container-runtime
|
|
ENV NVIDIA_VISIBLE_DEVICES all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,graphics
|
|
|
|
# Default pyopengl to EGL for good headless rendering support
|
|
ENV PYOPENGL_PLATFORM egl
|
|
|
|
WORKDIR /app
|
|
RUN git clone "https://github.com/comfyanonymous/ComfyUI.git" ./ && \
|
|
git reset --hard 05cd00695a84cebd5603a31f665eb7301fba2beb
|
|
RUN pip install --no-cache -r requirements.txt
|
|
|
|
WORKDIR /app/custom_nodes/ComfyUI-3D-Pack/
|
|
COPY --chown=user:user requirements.txt ./
|
|
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
|
|
|
|
WORKDIR /app/custom_nodes/
|
|
RUN git clone "https://github.com/ltdrdata/ComfyUI-Impact-Pack" && \
|
|
cd ComfyUI-Impact-Pack && \
|
|
git reset --hard e1570e76799049fb0038f6232fd89800ab4565eb && \
|
|
pip install --no-cache -r requirements.txt
|
|
RUN git clone "https://github.com/kijai/ComfyUI-KJNodes" && \
|
|
cd ComfyUI-KJNodes && \
|
|
git reset --hard 95ae8b067a8aa5825471189954579ffa02aa9256 && \
|
|
pip install --no-cache -r requirements.txt
|
|
RUN git clone "https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite" && \
|
|
cd ComfyUI-VideoHelperSuite && \
|
|
git reset --hard f94597739f20dac331f9918a245149b6b00a60f2 && \
|
|
pip install --no-cache -r requirements.txt
|
|
|
|
COPY --chown=user:user ./ ComfyUI-3D-Pack/
|
|
|
|
WORKDIR /app
|
|
ENTRYPOINT [ "python", "main.py", "--listen", "0.0.0.0" ]
|