update

finetune/Dockerfile

@@ -371,6 +371,7 @@ git checkout .
 python setup.py bdist_wheel
 DS_BUILD_OPS=${DS_BUILD_OPS} pip install dist/deepspeed*.whl --force-reinstall
 # DS_BUILD_OPS=${DS_BUILD_OPS} pip install -r requirements/requirements.txt
+./install.sh ${DEEPSPEED_INSTALL_FLAGS} 
 # ./install.sh ${DEEPSPEED_INSTALL_FLAGS} --hostfile /job/hostfile # ./install.sh --allow_sudo --pip_sudo --no_clean --hostfile /path/to/your/hostfile
 EOT
 

finetune/docker-compose_pytorch2.3.yml

@@ -21,9 +21,9 @@ services:
       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.0"
+        PYTORCH_VERSION: "2.3.1"
         TORCHVISION_VERSION: "0.18.0"
-        TORCHAUDIO_VERSION: "2.3.0"
+        TORCHAUDIO_VERSION: "2.3.1"
         DS_BUILD_OPS: 1
         DS_BUILD_SPARSE_ATTN: 0
         DS_BUILD_FUSED_ADAM: 1