add ldh

finetune/docker-compose_ldh.yml

@@ -0,0 +1,41 @@
+
+services:
+  ldh-deepspeed-test:
+    build: 
+      context: .
+      dockerfile: Dockerfile.ldh
+    image: ldh/deepspeed:test
+    shm_size: '32gb'
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+    runtime: nvidia
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all
+      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
+    stdin_open: true
+    tty: true
+    privileged: true
+    cap_add:
+      - IPC_LOCK
+    devices:
+      - /dev/infiniband:/dev/infiniband
+    volumes:
+      - /mnt/local-nvme2:/root/data/local-nvme2
+      - /mnt/local-nvme:/root/data/local-nvme
+    # ports:
+    #   - "22242:22242"
+    #   - "5000:5000"
+    # networks:
+    #   - ldh_overlay_network
+    network_mode: host
+    # command: ["/usr/sbin/sshd", "-D"]
+    # command: ["/usr/sbin/sshd"]
+
+# networks:
+#   ldh_overlay_network:
+#     external: true