update

2024-06-22 07:33:12 +00:00
parent 6a8b0fe992
commit 4a779d06ff
2 changed files with 46 additions and 36 deletions
--- a/finetune/docker-compose_pytorch2.3.yml
+++ b/finetune/docker-compose_pytorch2.3.yml
@@ -34,7 +34,7 @@ services:
        CUDA: cu121
        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 
+        DCUTLASS_NVCC_ARCHS: "80;86;89;90;90a" # 90a for H100 GPU 89:GeForce RTX 4090 
    volumes:
      - ./src:/bbtft
    container_name: ubuntu-finetune