diff --git a/finetune/docker-compose_pytorch1.13.yml b/finetune/docker-compose_pytorch1.13.yml index c2469ab..a42c597 100644 --- a/finetune/docker-compose_pytorch1.13.yml +++ b/finetune/docker-compose_pytorch1.13.yml @@ -19,7 +19,7 @@ services: USE_ROCM: 0 USE_XPU: 0 CUDA: cu117 - CUDA_ARCH_LIST: "80;86;89;90" # for RTX 4090, all : "80;86;89;90" + CUDA_ARCH_LIST: "80;86" # for RTX 4090, all : "80;86;89;90" SETUPTOOLS_VERSION: "69.5.1" ROOT_PASSWD: "root" DCUTLASS_NVCC_ARCHS: "90a" # 90a for H100 ,89:GeForce RTX 4090