update

finetune/docker-compose_stack.yml

@@ -43,7 +43,7 @@ services:
           generic_resources:
             - discrete_resource_spec:
                 kind: nvidia
-                value: 8
+                value: 1
       placement:
         constraints: [node.platform.os == linux]
     cap_add:

finetune/docker-compose_stack1.yml

@@ -0,0 +1,27 @@
+version: '3.8'
+
+services:
+  ubuntu-finetune:
+    image: hotwa/deepspeed:pt23
+    ports:
+      - 3228:22
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all
+      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
+      - TMPDIR=/var/tmp
+    deploy:
+      replicas: 1
+      resources:
+        reservations:
+          generic_resources:
+            - discrete_resource_spec:
+                kind: gpus
+                value: 1
+      placement:
+        constraints: [node.platform.os == linux]
+    cap_add:
+      - IPC_LOCK
+
+networks:
+  default:
+    driver: overlay