From 5c6e2f44e530b7097c46ab644ae9595631046213 Mon Sep 17 00:00:00 2001
From: hotwa <pylyzeng@gmail.com>
Date: Fri, 21 Jun 2024 15:02:18 +0800
Subject: [PATCH] cuda 11.7 suport CUDA_ARCH_LIST: "80;86"

---
 finetune/docker-compose_pytorch1.13.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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