Compare commits
6 Commits
feb21eb9a0
..
main
| Author | SHA1 | Date | |
|---|---|---|---|
| b1ae60d36f | |||
| c80d24411a | |||
| c9f2e3bf2f | |||
| d5de4893ec | |||
| 0b68bc2db0 | |||
| 6266b91c6d |
+10
-10
@@ -5,20 +5,15 @@ FROM ubuntu:26.04
|
|||||||
ENV PIP_BREAK_SYSTEM_PACKAGES=1
|
ENV PIP_BREAK_SYSTEM_PACKAGES=1
|
||||||
ENV PYTHONUNBUFFERED=1
|
ENV PYTHONUNBUFFERED=1
|
||||||
|
|
||||||
# Copy compiled ROCm from builder
|
|
||||||
COPY --from=builder /opt/rocm /opt/rocm
|
|
||||||
|
|
||||||
RUN apt update -y && apt install -y --no-install-recommends \
|
RUN apt update -y && apt install -y --no-install-recommends \
|
||||||
rocm-dev \
|
rocm \
|
||||||
rocm-libs \
|
|
||||||
rocminfo \
|
|
||||||
python3-pip \
|
python3-pip \
|
||||||
python3-dev \
|
python3-dev \
|
||||||
python3-torch-rocm \
|
python3-torch-rocm \
|
||||||
python3-torchaudio \
|
python3-torchaudio \
|
||||||
git \
|
git \
|
||||||
libgl1 \
|
libgl1 \
|
||||||
libglib2.0-0 \
|
libglib2.0-bin \
|
||||||
&& rm -rf /var/lib/apt/lists/*
|
&& rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
@@ -26,9 +21,14 @@ WORKDIR /app
|
|||||||
# Clone ComfyUI
|
# Clone ComfyUI
|
||||||
RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
|
RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
|
||||||
|
|
||||||
# Install ComfyUI dependencies
|
# Create sub dependancy
|
||||||
RUN grep -vE 'torch|torchaudio|torchvision' requirements.txt > req_no_torch.txt \
|
RUN grep -vE 'torch|nvidia|cuda|triton|kornia|spandrel' requirements.txt > req_no_torch.txt
|
||||||
&& pip3 install --no-cache-dir -r req_no_torch.txt
|
|
||||||
|
# Install non torch dependant requirements
|
||||||
|
RUN pip3 install --no-cache-dir --break-system-packages -r req_no_torch.txt
|
||||||
|
|
||||||
|
# Manually install the sensitive ones with --no-deps
|
||||||
|
RUN pip3 install --no-deps --break-system-packages torchsde kornia spandrel
|
||||||
|
|
||||||
# Setup Manager
|
# Setup Manager
|
||||||
WORKDIR /app/custom_nodes
|
WORKDIR /app/custom_nodes
|
||||||
|
|||||||
Reference in New Issue
Block a user