first add

docker-compose_stack1.yml

@@ -0,0 +1,37 @@
+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: "NVIDIA-GPU"
+                value: 1
+      placement:
+        constraints:
+          - node.labels.gpu == true
+    cap_add:
+      - IPC_LOCK
+
+networks:
+  default:
+    driver: overlay
+
+# 为节点添加标签：
+# docker node ls
+
+
+# docker node update --label-add gpu=true node1
+
+# docker stack deploy -c docker-compose.yml rdma_stack
+