From 660c098da05dd24febee45111b9117c7a75c2ed2 Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Fri, 2 Jun 2023 21:29:00 +0200
Subject: [PATCH] Change capabilities in docker-compose (#497)

#479
---
 docker-compose.yml                | 2 +-
 services/AUTOMATIC1111/Dockerfile | 1 -
 services/comfy/Dockerfile         | 2 +-
 services/invoke/Dockerfile        | 2 +-
 4 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index ab5bf8c5a..20e98f72e 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -13,7 +13,7 @@ x-base_service: &base_service
           devices:
               - driver: nvidia
                 device_ids: ['0']
-                capabilities: [gpu]
+                capabilities: [compute, utility]
 
 name: webui-docker
 
diff --git a/services/AUTOMATIC1111/Dockerfile b/services/AUTOMATIC1111/Dockerfile
index 173b356a0..6ef8fda10 100644
--- a/services/AUTOMATIC1111/Dockerfile
+++ b/services/AUTOMATIC1111/Dockerfile
@@ -88,7 +88,6 @@ RUN \
   git config --global --add safe.directory '*'
 
 WORKDIR ${ROOT}
-ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
 ENV NVIDIA_VISIBLE_DEVICES=all
 ENV CLI_ARGS=""
 EXPOSE 7860
diff --git a/services/comfy/Dockerfile b/services/comfy/Dockerfile
index ac0556fc3..ddf60fb20 100644
--- a/services/comfy/Dockerfile
+++ b/services/comfy/Dockerfile
@@ -38,7 +38,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 COPY . /docker/
 RUN cp /docker/extra_model_paths.yaml ${ROOT}
 
-ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility NVIDIA_VISIBLE_DEVICES=all
+ENV NVIDIA_VISIBLE_DEVICES=all
 ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
 EXPOSE 7860
 ENTRYPOINT ["/docker/entrypoint.sh"]
diff --git a/services/invoke/Dockerfile b/services/invoke/Dockerfile
index e3466d131..3bc90b1c6 100644
--- a/services/invoke/Dockerfile
+++ b/services/invoke/Dockerfile
@@ -42,7 +42,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 
 COPY . /docker/
 
-ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility NVIDIA_VISIBLE_DEVICES=all
+ENV NVIDIA_VISIBLE_DEVICES=all
 ENV PYTHONUNBUFFERED=1 PRELOAD=false HF_HOME=/root/.cache/huggingface CONFIG_DIR=/data/config/invoke CLI_ARGS=""
 EXPOSE 7860