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 PYTHONUNBUFFERED=1
|
||||
|
||||
# Copy compiled ROCm from builder
|
||||
COPY --from=builder /opt/rocm /opt/rocm
|
||||
|
||||
RUN apt update -y && apt install -y --no-install-recommends \
|
||||
rocm-dev \
|
||||
rocm-libs \
|
||||
rocminfo \
|
||||
rocm \
|
||||
python3-pip \
|
||||
python3-dev \
|
||||
python3-torch-rocm \
|
||||
python3-torchaudio \
|
||||
git \
|
||||
libgl1 \
|
||||
libglib2.0-0 \
|
||||
libglib2.0-bin \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
WORKDIR /app
|
||||
@@ -26,9 +21,14 @@ WORKDIR /app
|
||||
# Clone ComfyUI
|
||||
RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
|
||||
|
||||
# Install ComfyUI dependencies
|
||||
RUN grep -vE 'torch|torchaudio|torchvision' requirements.txt > req_no_torch.txt \
|
||||
&& pip3 install --no-cache-dir -r req_no_torch.txt
|
||||
# Create sub dependancy
|
||||
RUN grep -vE 'torch|nvidia|cuda|triton|kornia|spandrel' requirements.txt > 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
|
||||
WORKDIR /app/custom_nodes
|
||||
|
||||
Reference in New Issue
Block a user