Example configuration for ROCm image

[johncadengo]

Hi. Thanks for all of your hard work. Love the AI-Dock collection.

I just cloned the repository and am working off of the default Docker compose file:

services:
  supervisor:
    platform: linux/amd64
    build:
      context: ./build
      args:
        PYTHON_VERSION: ${PYTHON_VERSION:-3.10}
        PYTORCH_VERSION: ${PYTORCH_VERSION:-2.4.1}
        COMFYUI_BUILD_REF: ${COMFYUI_BUILD_REF:-}
        # Base on Python image which is base + python + jupyter
        IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/python:${PYTHON_VERSION:-3.10}-v2-cuda-12.1.1-base-22.04}
      tags:
        - "ghcr.io/ai-dock/comfyui:${IMAGE_TAG:-pytorch-${PYTORCH_VERSION:-2.3.0}-py3.10-v2-cuda-12.1.1-base-22.04}"
        
    image: ghcr.io/ai-dock/comfyui:${IMAGE_TAG:-pytorch-${PYTORCH_VERSION:-2.3.0}-py3.10-v2-cuda-12.1.1-base-22.04}

Is there a way to use .env variables to support the ROCm image or should I just edit the image to point to the one I want?

I'm wondering because there are a few args that depend on the python and pytorch versions. The pytorch version in the default Docker compose file refers to the CUDA version of pytorch.

[johncadengo]

This might be helpful for someone else, but this is what I'm trying right now:

In my .env file:

IMAGE_BASE:ghcr.io/ai-dock/comfyui:v2-rocm-6.0-runtime-22.04-v0.2.7
IMAGE_TAG:v2-rocm-6.0-runtime-22.04-v0.2.7

I got the image base and image tag from browsing ghcr.io, which is a link I found in the README: https://github.com/ai-dock/comfyui/pkgs/container/comfyui

[johncadengo]

Here is the error I'm getting, with the .env file as stated above.

[+] Building 114.8s (10/12)                                                                                                                   docker:default
 => [supervisor internal] load build definition from Dockerfile                                                                                         0.0s
 => => transferring dockerfile: 1.61kB                                                                                                                  0.0s
 => [supervisor internal] load metadata for ghcr.io/ai-dock/comfyui:v2-rocm-6.0-runtime-22.04-v0.2.7                                                    0.5s
 => [supervisor internal] load .dockerignore                                                                                                            0.0s
 => => transferring context: 2B                                                                                                                         0.0s
 => [supervisor internal] load build context                                                                                                            0.0s
 => => transferring context: 8.94kB                                                                                                                     0.0s
 => CACHED [supervisor 1/8] FROM ghcr.io/ai-dock/comfyui:v2-rocm-6.0-runtime-22.04-v0.2.7@sha256:a1e4ba44ad4d8d4babba11653eba4befac5c00f10c7bb10ba6496  0.0s
 => [supervisor 2/8] COPY --chown=0:1111 ./COPY_ROOT_0/ /                                                                                               0.1s
 => [supervisor 3/8] RUN set -eo pipefail && /opt/ai-dock/bin/build/layer0/init.sh | tee /var/log/build.log                                           111.2s
 => [supervisor 4/8] RUN echo "bust cache"                                                                                                              0.3s 
 => [supervisor 5/8] COPY --chown=0:1111 ./COPY_ROOT_1/ /                                                                                               0.1s 
 => ERROR [supervisor 6/8] RUN set -eo pipefail && /opt/ai-dock/bin/build/layer1/init.sh | tee -a /var/log/build.log                                    2.4s 
------                                                                                                                                                       
 > [supervisor 6/8] RUN set -eo pipefail && /opt/ai-dock/bin/build/layer1/init.sh | tee -a /var/log/build.log:                                               
0.647 Reading package lists...                                                                                                                               
1.269 Building dependency tree...                                                                                                                            
1.380 Reading state information...                                                                                                                           
1.476 libmagic1 is already the newest version (1:5.41-3ubuntu0.1).                                                                                           
1.476 0 upgraded, 0 newly installed, 0 to remove and 103 not upgraded.
1.712 Requirement already satisfied: aiocache in /opt/environments/python/api/lib/python3.10/site-packages (from -r /opt/ai-dock/api-wrapper/requirements.txt (line 1)) (0.12.3)

[...]

2.394 Stored environment variable 'COMFYUI_BUILD_REF': v0.3.22
2.398 fatal: destination path 'ComfyUI' already exists and is not an empty directory.
------
failed to solve: process "/bin/bash -c set -eo pipefail && /opt/ai-dock/bin/build/layer1/init.sh | tee -a /var/log/build.log" did not complete successfully: exit code: 128

To troubleshoot, I also ran it as docker compose build --no-cache and got the same error.

[johncadengo]

I also looked up the latest ROCm tag and tried that: v2-rocm-6.0-runtime-22.04-v0.2.2.

https://github.com/ai-dock/comfyui/pkgs/container/comfyui/279835212?tag=v2-rocm-6.0-runtime-22.04-v0.2.2

I still got the same error.