Split out Cuda and Rocm | Update

2026-04-16 17:03:53 -07:00
parent 3b94f7238d
commit d390f36947
6 changed files with 186 additions and 41 deletions
@@ -0,0 +1,44 @@
+# Use the official ROCm base image
+FROM rocm/dev-ubuntu-24.04:6.2
+
+# Set environment variables for 7900 XTX (gfx1100)
+ENV HSA_OVERRIDE_GFX_VERSION=11.0.0
+ENV TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
+ENV PYTHONUNBUFFERED=1
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    python3-pip \
+    python3-venv \
+    git \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set working directory
+WORKDIR /app
+
+# Clone ComfyUI
+RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
+
+# Install PyTorch with ROCm 6.2 support (matching the base image)
+RUN pip3 install --no-cache-dir torch torchvision torchaudio \
+    --index-url https://download.pytorch.org/whl/rocm6.2
+
+# Install ComfyUI dependencies
+RUN pip3 install --no-cache-dir -r requirements.txt
+
+# Set custom Nodes
+WORKDIR /app/custom_nodes
+
+# Install ComfyUi Manager
+RUN git clone https://github.com/Comfy-Org/ComfyUI-Manager.git
+
+# Go back to root
+WORKDIR /app
+
+# Expose the default ComfyUI port
+EXPOSE 8188
+
+# Launch ComfyUI
+CMD ["python3", "main.py", "--listen", "0.0.0.0"]