update

finetune/docker-compose_ldh.yml

@@ -4,6 +4,9 @@ services:
     build: 
       context: .
       dockerfile: Dockerfile.ldh
+      args:
+        HTTP_PROXY: "http://127.0.0.1:15777"
+        HTTPS_PROXY: "http://127.0.0.1:15777"
     image: ldh/deepspeed:test
     shm_size: '32gb'
     deploy:
@@ -13,28 +16,30 @@ services:
             - driver: nvidia
               count: all
               capabilities: [gpu]
-    runtime: nvidia
+    #runtime: nvidia
     environment:
       - NVIDIA_VISIBLE_DEVICES=all
       - NVIDIA_DRIVER_CAPABILITIES=compute,utility
+        #- CUTLASS_PATH="/opt/cutlass"
+        #- CUDA_HOME="/usr/local/cuda"
+        #- PATH="${CUDA_HOME}/bin:${PATH}"
+        #- LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"
     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"
+      # - /mnt/local-nvme2:/root/workspace
+      # - /dev/infiniband:/dev/infiniband
+    #  - /mnt/local-nvme:/root/
+    ports:
+      - "22242:22242"
+      - "5000:5000"
     # networks:
     #   - ldh_overlay_network
     network_mode: host
-    # command: ["/usr/sbin/sshd", "-D"]
-    # command: ["/usr/sbin/sshd"]
+    command: ["/usr/sbin/sshd", "-D"]
 
 # networks:
 #   ldh_overlay_network: