forked from Stability-AI/stable-fast-3d
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDockerfile
128 lines (101 loc) · 5.16 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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
################################################################################
# 用于构建镜像 'yanwk/comfyui-boot:cu124-cn' 的 Dockerfile
# 为适应国内网络环境，运行时的下载地址均改用国内源。
# 但需注意，构建本镜像时仍需要流畅的网络环境（完全走国内源请参考另一 Dockerfile）。
#
# 使用 CUDA 12.4, Python 3.12, GCC 13
# 容器内将以 root 用户运行，以便于 rootless 部署
################################################################################
FROM docker.io/opensuse/tumbleweed:latest
LABEL maintainer="YAN Wenkun <[email protected]>"
RUN set -eu
ARG HUGGINGFACE_TOKEN1
ENV HUGGINGFACE_TOKEN=$HUGGINGFACE_TOKEN1
ENV USE_CUDA=1
# USER root
# VOLUME /root
# WORKDIR /root
EXPOSE 7860
COPY . .
################################################################################
# Python 及工具
# 利用 openSUSE 软件仓库的 PIP 包，以确保兼容性以及更多的系统级支持。后续仍可使用 PIP 更新。
RUN --mount=type=cache,target=/var/cache/zypp \
zypper addrepo --check --refresh --priority 90 \
'https://ftp.gwdg.de/pub/linux/misc/packman/suse/openSUSE_Tumbleweed/Essentials/' packman-essentials \
&& zypper --gpg-auto-import-keys \
install --no-confirm --auto-agree-with-licenses \
python310-devel \
python310-pip \
python310-wheel \
python310-setuptools \
git \
make \
ninja \
which \
#&& rm /usr/lib64/python3.10/EXTERNALLY-MANAGED \
&& update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 100
################################################################################
# GCC 13
# 与 CUDA 12.4 兼容
RUN --mount=type=cache,target=/var/cache/zypp \
zypper --gpg-auto-import-keys \
install --no-confirm --auto-agree-with-licenses \
gcc12 \
gcc12-c++ \
cpp12 \
&& update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++-12 90 \
&& update-alternatives --install /usr/bin/cc cc /usr/bin/gcc-12 90 \
&& update-alternatives --install /usr/bin/cpp cpp /usr/bin/cpp-12 90 \
&& update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-12 90 \
&& update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-12 90 \
&& update-alternatives --install /usr/bin/gcc-ar gcc-ar /usr/bin/gcc-ar-12 90 \
&& update-alternatives --install /usr/bin/gcc-nm gcc-nm /usr/bin/gcc-nm-12 90 \
&& update-alternatives --install /usr/bin/gcc-ranlib gcc-ranlib /usr/bin/gcc-ranlib-12 90 \
&& update-alternatives --install /usr/bin/gcov gcov /usr/bin/gcov-12 90 \
&& update-alternatives --install /usr/bin/gcov-dump gcov-dump /usr/bin/gcov-dump-12 90 \
&& update-alternatives --install /usr/bin/gcov-tool gcov-tool /usr/bin/gcov-tool-12 90
################################################################################
# Python 包
# PyTorch, xFormers
RUN --mount=type=cache,target=/root/.cache/pip \
pip list \
&& pip install -U setuptools==69.5.1 \
&& pip install torch==2.4.1 \
&& pip install numpy==1.26.4
# 绑定环境变量 (依赖库 .so 文件)
# ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}\
# :/usr/local/lib64/python3.12/site-packages/torch/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/cuda_cupti/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/cuda_runtime/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/cudnn/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/cufft/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/cublas/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/cuda_nvrtc/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/curand/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/cusolver/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/cusparse/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/nccl/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/nvjitlink/lib\
# :/usr/local/lib/python3.12/site-packages/nvidia/nvtx/lib"
# 1. 安装 ComfyUI 及扩展的依赖项
# 2. 处理 ONNX Runtime 报错 "missing CUDA provider"，并添加 CUDA 12 支持，参考： https://onnxruntime.ai/docs/install/
# 3. 接上，处理 MediaPipe's 的依赖项错误（需要 protobuf<4）
RUN --mount=type=cache,target=/root/.cache/pip \
pip install -r requirements.txt
################################################################################
# 使用国内下载源
# RUN --mount=type=cache,target=/var/cache/zypp \
# zypper modifyrepo --disable --all \
# && zypper addrepo --check --refresh --gpgcheck \
# 'https://mirrors.tuna.tsinghua.edu.cn/opensuse/tumbleweed/repo/oss/' mirror-oss \
# && zypper addrepo --check --refresh --gpgcheck \
# 'https://mirrors.tuna.tsinghua.edu.cn/opensuse/tumbleweed/repo/non-oss/' mirror-non-oss \
# && zypper addrepo --check --refresh --priority 90 \
# 'https://mirrors.tuna.tsinghua.edu.cn/packman/suse/openSUSE_Tumbleweed/Essentials/' mirror-packman-essentials
# ENV PIP_INDEX_URL="https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
# ENV HF_ENDPOINT="https://hf-mirror.com"
################################################################################
RUN python3 ./download.py
ENV CLI_ARGS=""
CMD ["python3","./gradio_app.py"]