update

finetune/docker-compose_pytorch2.3.yml

@@ -19,11 +19,11 @@ services:
       context: .
       dockerfile: Dockerfile
       args: # PyTorch版本、Python版本与pytorch_lightning版本的对应关系表 https://blog.csdn.net/qq_41813454/article/details/137421822
-        PYTHON_VERSION: 3.10
-        CUDA_VERSION: 12.1.0
-        PYTORCH_VERSION: 2.3.1
-        TORCHVISION_VERSION: 0.18.0
-        TORCHAUDIO_VERSION: 2.3.0
+        PYTHON_VERSION: "3.10"
+        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: 1
@@ -32,7 +32,7 @@ services:
         USE_ROCM: 0
         USE_XPU: 0
         CUDA: cu121
-        CUDA_ARCH_LIST: "80;86;89;90;90a" # for RTX 4090, all : "80;86;89;90"
+        CUDA_ARCH_LIST: "80;86;89;90" # for RTX 4090, all : "80;86;89;90"
         SETUPTOOLS_VERSION: "69.5.1"
         DCUTLASS_NVCC_ARCHS: "90a" # 90a for H100 GPU 89:GeForce RTX 4090 
     volumes:
@@ -44,7 +44,7 @@ services:
     image: hotwa/deepspeed:pt23
     shm_size: '32gb'
     ports:
-      - 3227:2222
+      - 3228:2222
     environment:
       - NVIDIA_VISIBLE_DEVICES=all
       - NVIDIA_DRIVER_CAPABILITIES=compute,utility