From e0bfbba17e04713925e2c306679b0f040614f50a Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Mon, 2 Sep 2024 11:27:20 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0photo=5Freview=5F6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker-compose.yml | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/docker-compose.yml b/docker-compose.yml
index 4951fda..fd4dabc 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -143,5 +143,23 @@ services:
         reservations:
           devices:
             - device_ids: [ "1" ]
+              capabilities: [ "gpu" ]
+              driver: "nvidia"
+
+  photo_review_6:
+    <<: *template
+    container_name: photo_review_6
+    hostname: photo_review_6
+    volumes:
+      - ./log:/app/log
+      - ./model:/app/model
+    depends_on:
+      - photo_mask_2
+    command: [ "photo_review.py" ]
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - device_ids: [ "0", "1" ]
               capabilities: [ "gpu" ]
               driver: "nvidia"
\ No newline at end of file