From e3f2375e4c6fe57d96bdbf1499b73be377a46719 Mon Sep 17 00:00:00 2001
From: Derek Holloway <derek@mistox.com>
Date: Thu, 16 Apr 2026 19:15:17 -0700
Subject: [PATCH] build with no libraries and pass in the device via compose

---
 .gitea/workflows/build.yaml |  6 +++---
 comfyui-rocm-dockerfile     | 14 +++++---------
 2 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/.gitea/workflows/build.yaml b/.gitea/workflows/build.yaml
index 77ec378..0f8bad8 100644
--- a/.gitea/workflows/build.yaml
+++ b/.gitea/workflows/build.yaml
@@ -23,7 +23,7 @@ jobs:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
-      - name: build and push comfyui
+      - name: build and push comfyui-cuda
         run: |
           docker buildx build \
           --platform=linux/arm64 \
@@ -48,7 +48,7 @@ jobs:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
-      - name: build and push comfyui
+      - name: build and push comfyui-rocm
         run: |
           docker buildx build \
           --platform=linux/arm64 \
@@ -73,7 +73,7 @@ jobs:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
-      - name: build and push comfyui
+      - name: build and push uvicorn
         run: |
           docker buildx build \
           --platform=linux/arm64 \
diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 9e2e8c9..575cbba 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -1,19 +1,15 @@
 # Use the official ROCm base image
-FROM rocm/dev-ubuntu-24.04:6.2
+FROM ubuntu:24.04
 
 # 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 apk add --no-cache \
-    python3 \
-    py3-pip \
-    git \
-    mesa-gles \
-    glib \
-    && rm -rf /var/cache/apk/*
+# Install basic Python and Graphics libs
+RUN apt update -y && apt install -y \
+    python3-pip python3-dev git libgl1-mesa-glx libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
 
 # Set working directory
 WORKDIR /app