Merge pull request #1134 from exadel-inc/EFRS-1397_update_docker_comp…

…ose_GPU EFRS-1397 Update docker compose gpu
exadel-inc · Aug 1, 2023 · 7a441f8 · 7a441f8
2 parents c985c8b + 59deac3
commit 7a441f8
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 3 deletions.
diff --git a/custom-builds/Mobilenet-gpu/docker-compose.yml b/custom-builds/Mobilenet-gpu/docker-compose.yml
@@ -72,10 +72,16 @@ services:
     image: ${registry}compreface-core:${CORE_VERSION}
     restart: always
     container_name: "compreface-core"
-    runtime: nvidia
     environment:
       - ML_PORT=3000
       - IMG_LENGTH_LIMIT=${max_detect_size}
       - UWSGI_PROCESSES=${uwsgi_processes:-1}
       - UWSGI_THREADS=${uwsgi_threads:-1}
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
 
diff --git a/custom-builds/SubCenter-ArcFace-r100-gpu/docker-compose.yml b/custom-builds/SubCenter-ArcFace-r100-gpu/docker-compose.yml
@@ -72,10 +72,16 @@ services:
     image: ${registry}compreface-core:${CORE_VERSION}
     restart: always
     container_name: "compreface-core"
-    runtime: nvidia
     environment:
       - ML_PORT=3000
       - IMG_LENGTH_LIMIT=${max_detect_size}
       - UWSGI_PROCESSES=${uwsgi_processes:-2}
       - UWSGI_THREADS=${uwsgi_threads:-1}
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
 
diff --git a/dev/docker-compose-gpu.yml b/dev/docker-compose-gpu.yml
@@ -91,7 +91,6 @@ services:
     container_name: "compreface-core"
     ports:
       - "3300:3000"
-    runtime: nvidia
     build:
       context: ../embedding-calculator
       args:
@@ -104,3 +103,10 @@ services:
       - ML_PORT=3000
       - UWSGI_PROCESSES=${uwsgi_processes:-1}
       - UWSGI_THREADS=${uwsgi_threads:-1}
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]