update

megadna/docker-compose_pytorch2.3.yml

@@ -0,0 +1,45 @@
+version: '3.8'
+
+services:
+  ubuntu-finetune:
+    build: 
+      context: .
+      dockerfile: Dockerfile
+      args: # PyTorch版本、Python版本与pytorch_lightning版本的对应关系表 https://blog.csdn.net/qq_41813454/article/details/137421822
+        CUDA_VERSION: 12.1.0
+        PYTORCH_VERSION: 2.3.0
+        TORCHVISION_VERSION: 0.18.0
+        TORCHAUDIO_VERSION: 2.3.0
+        DS_BUILD_OPS: 1
+        DS_BUILD_SPARSE_ATTN: 0
+        DS_BUILD_FUSED_ADAM: 0
+        DS_BUILD_CPU_ADAM: 0
+        CUDA: cu121
+        CUDA_ARCH_LIST: "80;86;89;90" # for RTX 4090, all : "80;86;89;90"
+        SETUPTOOLS_VERSION: "69.5.1"
+    volumes:
+      - ./src:/bbtft
+    container_name: ubuntu-finetune
+    pull_policy: if_not_present
+    tty: true
+    restart: unless-stopped
+    image: hotwa/deepspeed:test
+    shm_size: '32gb'
+    ports:
+      - 3227:2222
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all
+      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
+    networks:
+      - network_finetune
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+
+networks:
+  network_finetune:
+    name: network_finetune