File size: 1,464 Bytes
70221ee 3ed3735 70221ee df9ec43 70221ee ec497e7 70221ee bffdf27 70221ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
ARG CUDA_VERSION="12.1.1"
ARG CUDNN_VERSION="8"
ARG UBUNTU_VERSION="22.04"
ARG DOCKER_FROM=nvidia/cuda:$CUDA_VERSION-cudnn$CUDNN_VERSION-devel-ubuntu$UBUNTU_VERSION
# Base NVidia CUDA Ubuntu image
FROM $DOCKER_FROM AS base
# Install Python plus openssh, which is our minimum set of required packages.
RUN apt-get update -y && \
apt-get install -y python3 python3-pip python3-venv && \
apt-get install -y --no-install-recommends openssh-server openssh-client git git-lfs wget vim zip unzip curl && \
python3 -m pip install --upgrade pip && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
# Install nginx
RUN apt-get update && \
apt-get install -y nginx
# Copy the 'default' configuration file to the appropriate location
COPY default /etc/nginx/sites-available/default
ENV PATH="/usr/local/cuda/bin:${PATH}"
# Install pytorch
ARG PYTORCH="2.4.0"
ARG CUDA="121"
RUN pip3 install --no-cache-dir -U torch==$PYTORCH torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu$CUDA
WORKDIR /workspace
# Clone the git repo and install requirements in the same RUN command to ensure they are in the same layer
RUN git clone https://github.com/comfyanonymous/ComfyUI.git && \
cd ComfyUI && \
pip3 install -r requirements.txt
COPY --chmod=755 start.sh /start.sh
COPY --chmod=755 download_models.sh /download_models.sh
# Add Jupyter Notebook
RUN pip3 install jupyterlab
EXPOSE 8888
EXPOSE 8188
CMD [ "/start.sh" ] |