Init docker support (Nvidia and AMD)

docker-compose.yml

@@ -0,0 +1,67 @@
+version: "3.9"
+name: "comfyui"
+
+
+x-base_service: &base_service
+  image: comfyui:1
+  build:
+    context: .
+    dockerfile: dockerfiles/default.Dockerfile
+
+  environment:
+    - CLI_ARGS=--listen --port 8188
+
+  ports:
+    - "8188:8188"
+  volumes:
+    - ./models:/opt/comfy_ui/models
+    - ./output:/opt/comfy_ui/output
+    - ./input:/opt/comfy_ui/input
+    - ./custom_nodes:/opt/comfy_ui/custom_nodes
+
+x-nvidia_docker: &nvidia_docker
+  deploy:
+    resources:
+      reservations:
+        devices:
+          - driver: nvidia
+            device_ids: ["0"]
+            capabilities: [gpu]
+
+x-base_service_amd: &amd_docker
+  build:
+    context: .
+    dockerfile: dockerfiles/amd.Dockerfile
+
+  group_add:
+    - video
+  devices:
+    - "/dev/dri"
+    - "/dev/kfd"
+
+
+services:
+  nvidia:
+    <<: *base_service
+    <<: *nvidia_docker
+    profiles: ["nvidia"]
+
+  amd:
+    <<: *base_service
+    <<: *amd_docker
+    profiles: ["amd"]
+
+  amd6600:
+    <<: *base_service
+    <<: *amd_docker
+    profiles: ["amd6600"]
+
+    environment:
+      - HSA_OVERRIDE_GFX_VERSION=10.3.0
+      - CLI_ARGS=--listen --port 8188
+
+  cpu:
+    <<: *base_service
+    profiles: ["cpu"]
+    environment:
+      - CLI_ARGS=--listen --port 8188 --cpu

dockerfiles/amd.Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.10-slim
+
+
+WORKDIR /opt/comfy_ui
+
+
+RUN PIP_NO_CACHE_DIR=1 pip install --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 \
+torch \
+torchvision \
+torchaudio
+
+COPY . /opt/comfy_ui
+RUN pip install -r requirements.txt
+
+
+ENV CLI_ARGS=""
+CMD python -u main.py ${CLI_ARGS}

dockerfiles/default.Dockerfile

@@ -0,0 +1,19 @@
+FROM python:3.10-slim
+
+
+WORKDIR /opt/comfy_ui
+
+
+RUN PIP_NO_CACHE_DIR=1 pip install --extra-index-url https://download.pytorch.org/whl/cu117 \
+torch==1.13.1+cu117 \
+torchvision \
+torchaudio \
+xformers \
+triton
+
+COPY . /opt/comfy_ui
+RUN pip install -r requirements.txt
+
+
+ENV CLI_ARGS=""
+CMD python -u main.py ${CLI_ARGS}