forked from comfyanonymous/ComfyUI
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
112 lines (90 loc) · 3.85 KB
/
Dockerfile
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
# Stage 1: Production stage
FROM nvidia/cuda:12.1.1-runtime-ubuntu22.04 as runtime
# Configure shell
SHELL ["/bin/bash", "-o", "pipefail", "-c"]
ENV SHELL=/bin/bash
ENV PYTHONUNBUFFERED=1
# Set the working directory in the docker container
WORKDIR /gen-server
# Configure apt-get to automatically use noninteractive settings
ENV DEBIAN_FRONTEND=noninteractive
# Install system libraries
RUN apt-get update --yes && \
apt-get upgrade --yes && \
apt-get install --yes --no-install-recommends git wget curl bash libgl1 fuse software-properties-common openssh-server nginx rsync dos2unix ffmpeg libmagic1 && \
add-apt-repository ppa:deadsnakes/ppa && \
apt-get --yes --no-install-recommends install python3.11-dev python3.11-venv && \
echo "en_US.UTF-8 UTF-8" > /etc/locale.gen
# Install GCC and ONNX Runtime
RUN apt-get install --yes libcurand10 gcc g++ && \
wget https://github.com/microsoft/onnxruntime/releases/download/v1.16.3/onnxruntime-linux-x64-gpu-1.16.3.tgz && \
tar zxvf onnxruntime-linux-x64-gpu-1.16.3.tgz && \
rm onnxruntime-linux-x64-gpu-1.16.3.tgz && \
apt-get autoremove -y && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
# Set up Python and pip
RUN ln -s /usr/bin/python3.11 /usr/bin/python && \
rm /usr/bin/python3 && \
ln -s /usr/bin/python3.11 /usr/bin/python3 && \
curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && \
python get-pip.py
RUN python -m venv /venv
ENV PATH="/venv/bin:$PATH"
# Install this project's python packages
RUN pip install --upgrade --no-cache-dir pip && \
pip install --upgrade setuptools && \
pip install --upgrade wheel && \
pip install --upgrade --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
COPY requirements.txt requirements-new.txt ./
RUN pip install --no-cache-dir -r requirements.txt \
&& pip install --no-cache-dir -r requirements-new.txt
# Set up Jupyter Notebook
RUN jupyter contrib nbextension install --user && \
jupyter nbextension enable --py widgetsnbextension
# Folders / files in .dockerignore will not be included
# Copy the necessary parts of Comfy Inference API.
COPY . .
# COPY ./app ./app
# COPY ./build_files ./build_files
# COPY ./comfy ./comfy
# COPY ./comfy_extras ./comfy_extras
# COPY ./notebooks ./notebooks
# COPY ./script_examples ./script_examples
# Copy over our UI build files
# COPY --from=builder ./web ./web
# Setup NGINX Proxy
RUN mv ./build_files/nginx.conf /etc/nginx/nginx.conf
RUN mv ./build_files/readme.html /usr/share/nginx/html/readme.html
# Use dos2unix to ensure line-endings are unix-style
# TO DO: if we build these directories into Comfy Inference API, we can remove this step
RUN mv ./build_files/model_paths/symlinks.txt .
RUN dos2unix ./symlinks.txt
# Copying the catfs precompiled binary to container
COPY catfs /usr/local/bin
# Installing rsfw-cache
# WORKDIR /opt
#RUN git clone https://github.com/m-arbaro/rsfw-cache
# WORKDIR /gen-server
# Creating mountpoint for tmpfs:
# RUN mkdir /usr/share/memory
# Clone Custom-Nodes into container and install their dependencies
RUN mkdir -p /usr/share/custom_nodes && \
mv ./build_files/custom_nodes/repos.txt /usr/share/custom_nodes/
RUN mv ./build_files/custom_nodes/pull_custom_nodes.sh /usr/local/bin/pull_custom_nodes.sh && \
dos2unix /usr/local/bin/pull_custom_nodes.sh && \
/usr/local/bin/pull_custom_nodes.sh > /var/log/pull_custom_nodes.log
RUN mv ./build_files/model_paths ./model_paths
# This will be our start script
RUN mv ./build_files/start.sh . && \
rm -rf ./build_files
RUN dos2unix ./start.sh && \
chmod +x ./start.sh
# Set default values for environment variables
# ENV APP_HOST=0.0.0.0
# ENV APP_PORT=8080
# Create user group so we don't have to run as root
# RUN useradd -l -M appuser
# USER appuser
# Run the Comfy Generation Server, Jupyter Notebook, and NGINX Proxy
CMD ["/gen-server/start.sh"]