为照片涂抹编写docker

docker-compose.yml

@@ -0,0 +1,47 @@
+version: '3.9'
+
+services:
+  photo_mask_1:
+    container_name: photo_mask_1
+    image: photo_mask
+    build:
+      context: .
+    restart: always
+    command: [ "photo_mask.py", "--clean", "True" ]
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - device_ids: [ "1" ]
+              capabilities: [ "gpu" ]
+              driver: "nvidia"
+
+  photo_mask_2:
+    container_name: photo_mask_2
+    image: photo_mask
+    restart: always
+    depends_on:
+      - photo_mask_1
+    command: [ "photo_mask.py" ]
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - device_ids: [ "1" ]
+              capabilities: [ "gpu" ]
+              driver: "nvidia"
+
+  photo_mask_3:
+    container_name: photo_mask_3
+    image: photo_mask
+    restart: always
+    depends_on:
+      - photo_mask_2
+    command: [ "photo_mask.py" ]
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - device_ids: [ "1" ]
+              capabilities: [ "gpu" ]
+              driver: "nvidia"