From 6df4ef5f1762a3b251ac330c8901d679478cf7a2 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Tue, 2 Jul 2024 07:23:21 +0000
Subject: [PATCH] update

---
 finetune/docker-compose_stack.yml  |  2 +-
 finetune/docker-compose_stack1.yml | 27 +++++++++++++++++++++++++++
 2 files changed, 28 insertions(+), 1 deletion(-)
 create mode 100644 finetune/docker-compose_stack1.yml

diff --git a/finetune/docker-compose_stack.yml b/finetune/docker-compose_stack.yml
index 5fecd98..52841bb 100644
--- a/finetune/docker-compose_stack.yml
+++ b/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:
diff --git a/finetune/docker-compose_stack1.yml b/finetune/docker-compose_stack1.yml
new file mode 100644
index 0000000..b8e4497
--- /dev/null
+++ b/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
\ No newline at end of file