udpate

2024-06-22 12:07:07 +00:00
parent aa447a150c
commit 273257efc3
3 changed files with 103 additions and 3 deletions
--- a/finetune/docker-compose_pytorch2.3.yml
+++ b/finetune/docker-compose_pytorch2.3.yml
@@ -35,6 +35,7 @@ services:
        CUDA_ARCH_LIST: "80;86;89;90" # for RTX 4090, all : "80;86;89;90"
        SETUPTOOLS_VERSION: "69.5.1"
        DCUTLASS_NVCC_ARCHS: "80;86;89;90;90a" # 90a for H100 GPU 89:GeForce RTX 4090 
+        DEEPSPEED_VERSION: "0.14.3"
        DEEPSPEED_INSTALL_FLAGS: "--allow_sudo --pip_sudo --verbose"
    volumes:
      - ./src:/bbtft