build with no libraries and pass in the device via compose
This commit is contained in:
@@ -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 \
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user