wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-15 13:02:35 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Update Dockerfile

Browse Source 
Change cuda from 12.9 to 12.8 due to pytorch only being maintained for 12.6 and 12.8
This commit is contained in:
Sublime 2025-11-02 20:50:37 -08:00 committed by GitHub
parent 6ad1f6c168
commit 4df6b8367a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
Dockerfile
View File

@ -40,13 +40,13 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
&& dpkg -i cuda-keyring_1.1-1_all.deb \ && dpkg -i cuda-keyring_1.1-1_all.deb \
&& apt-get update \ && apt-get update \
&& apt-get install -y --no-install-recommends \ && apt-get install -y --no-install-recommends \
cuda-nvcc-12-9 \ cuda-nvcc-12-8 \
cuda-cudart-dev-12-9 \ cuda-cudart-dev-12-8 \
libcusparse-dev-12-9 \ libcusparse-dev-12-8 \
libcublas-dev-12-9 \ libcublas-dev-12-8 \
libcurand-dev-12-9 \ libcurand-dev-12-8 \
libcusolver-dev-12-9 \ libcusolver-dev-12-8 \
libcufft-dev-12-9 \ libcufft-dev-12-8 \
nvidia-smi \ nvidia-smi \
&& rm -rf /var/lib/apt/lists/* \ && rm -rf /var/lib/apt/lists/* \
&& rm cuda-keyring_1.1-1_all.deb && rm cuda-keyring_1.1-1_all.deb
@ -58,12 +58,12 @@ RUN sed -i 's/extern __DEVICE_FUNCTIONS_DECL__ __device_builtin__ double
sed -i 's/extern __DEVICE_FUNCTIONS_DECL__ __device_builtin__ float cospif(float x);/extern __DEVICE_FUNCTIONS_DECL__ __device_builtin__ float cospif(float x) noexcept (true);/' /usr/local/cuda-12.9/include/crt/math_functions.h sed -i 's/extern __DEVICE_FUNCTIONS_DECL__ __device_builtin__ float cospif(float x);/extern __DEVICE_FUNCTIONS_DECL__ __device_builtin__ float cospif(float x) noexcept (true);/' /usr/local/cuda-12.9/include/crt/math_functions.h
# Set CUDA paths for entrypoint compilation # Set CUDA paths for entrypoint compilation
ENV CUDA_HOME=/usr/local/cuda-12.9 \ ENV CUDA_HOME=/usr/local/cuda-12.8 \
PATH=/usr/local/cuda-12.9/bin:${PATH} \ PATH=/usr/local/cuda-12.8/bin:${PATH} \
LD_LIBRARY_PATH=/usr/local/cuda-12.9/lib64 LD_LIBRARY_PATH=/usr/local/cuda-12.8/lib64
# Create symlink for compatibility # Create symlink for compatibility
RUN ln -sf /usr/local/cuda-12.9 /usr/local/cuda RUN ln -sf /usr/local/cuda-12.8 /usr/local/cuda
# Create runtime user/group # Create runtime user/group
RUN set -e; \ RUN set -e; \
@ -88,9 +88,9 @@ WORKDIR /app/ComfyUI
# Copy requirements with optional handling # Copy requirements with optional handling
COPY requirements.txt* ./ COPY requirements.txt* ./
# Core Python deps (torch CUDA 12.9, pin Triton, plus common deps) # Core Python deps (torch CUDA 12.8, pin Triton, plus common deps)
RUN python -m pip install --upgrade pip setuptools wheel \ RUN python -m pip install --upgrade pip setuptools wheel \
&& python -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu129 \ && python -m pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu128 \
&& python -m pip install "triton==3.4.0" \ && python -m pip install "triton==3.4.0" \
&& python -m pip install --prefer-binary cupy-cuda12x \ && python -m pip install --prefer-binary cupy-cuda12x \
&& if [ -f requirements.txt ]; then python -m pip install -r requirements.txt; fi \ && if [ -f requirements.txt ]; then python -m pip install -r requirements.txt; fi \