cuda 11.7 suport CUDA_ARCH_LIST: "80;86"

2024-06-21 15:02:18 +08:00
parent d351fe520a
commit 5c6e2f44e5
1 changed files with 1 additions and 1 deletions
--- 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