update

2024-06-21 15:15:15 +08:00
parent be77975ce2
commit 2620524636
2 changed files with 27 additions and 25 deletions
--- a/finetune/docker-compose_pytorch2.3.yml
+++ b/finetune/docker-compose_pytorch2.3.yml
@@ -32,7 +32,7 @@ services:
        USE_ROCM: 0
        USE_XPU: 0
        CUDA: cu121
-        CUDA_ARCH_LIST: "80;86;89;90" # for RTX 4090, all : "80;86;89;90"
+        CUDA_ARCH_LIST: "80;86;89;90;90a" # 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: