mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-21 23:41:28 +08:00
dockerfile updates
This commit is contained in:
+12
-6
@@ -1,6 +1,6 @@
|
||||
FROM nvcr.io/nvidia/pytorch:25.03-py3
|
||||
|
||||
ARG TZ="America/Los_Angeles"
|
||||
ENV TZ="Etc/UTC"
|
||||
|
||||
ENV PYTORCH_CUDA_ALLOC_CONF="backend:cudaMallocAsync"
|
||||
ENV UV_COMPILE_BYTECODE=1
|
||||
@@ -9,23 +9,29 @@ ENV UV_SYSTEM_PYTHON=1
|
||||
ENV UV_BREAK_SYSTEM_PACKAGES=1
|
||||
ENV PIP_DISABLE_PIP_VERSION_CHECK=1
|
||||
ENV PIP_NO_CACHE_DIR=1
|
||||
ENV DEBIAN_FRONTEND="noninteractive"
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
|
||||
ENV LANG=C.UTF-8
|
||||
ENV LC_ALL=C.UTF-8
|
||||
|
||||
# mitigates
|
||||
# RuntimeError: Failed to import transformers.generation.utils because of the following error (look up to see its traceback):
|
||||
# numpy.dtype size changed, may indicate binary incompatibility. Expected 96 from C header, got 88 from PyObject
|
||||
RUN echo "numpy<2" >> numpy-override.txt
|
||||
|
||||
# mitigates https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
|
||||
# mitigates AttributeError: module 'cv2.dnn' has no attribute 'DictValue' \
|
||||
# see https://github.com/facebookresearch/nougat/issues/40
|
||||
RUN pip install uv && uv --version && \
|
||||
apt-get update && apt-get install --no-install-recommends ffmpeg libsm6 libxext6 -y && \
|
||||
RUN apt-get update && \
|
||||
apt-get install --no-install-recommends -y tzdata ffmpeg libsm6 libxext6 && \
|
||||
pip install uv && uv --version && \
|
||||
apt-get purge -y --auto-remove tzdata && \
|
||||
rm -rf /var/lib/apt/lists/*
|
||||
|
||||
RUN uv pip uninstall --system $(pip list --format=freeze | grep opencv) && \
|
||||
rm -rf /usr/local/lib/python3.12/dist-packages/cv2/ && \
|
||||
uv pip install --no-build-isolation opencv-python-headless
|
||||
RUN uv pip install --overrides=numpy-override.txt "comfyui[attention,comfyui_manager]@git+https://github.com/hiddenswitch/ComfyUI.git" && \
|
||||
|
||||
RUN uv pip install --overrides=numpy-override.txt "comfyui[attention,comfyui_manager]@git+https://github.com/hiddenswitch/ComfyUI.git"
|
||||
|
||||
WORKDIR /workspace
|
||||
# addresses https://github.com/pytorch/pytorch/issues/104801
|
||||
|
||||
Reference in New Issue
Block a user