From adb3a10bdbb1ee6c4cec564c042c3d06eb08ff5e Mon Sep 17 00:00:00 2001
From: lingyuzeng <pylyzeng@gmail.com>
Date: Wed, 17 Jul 2024 13:03:54 +0800
Subject: [PATCH] TORCH_CUDA_ARCH_LIST: "8.0;8.6;8.9;9.0+PTX" add

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

diff --git a/finetune/docker-compose_update.yml b/finetune/docker-compose_update.yml
index 252bb96..cc24ed4 100644
--- a/finetune/docker-compose_update.yml
+++ b/finetune/docker-compose_update.yml
@@ -34,7 +34,7 @@ services:
         USE_XPU: 0
         CUDA: cu121
         CUDA_ARCH_LIST: "80;86;89;90" # for RTX 4090, all : "80;86;89;90"
-        TORCH_CUDA_ARCH_LIST: "8.0;8.6;8.9;9.0" # all "6.0;6.1;6.2;7.0;7.5;8.0;8.6;8.9;9.0"
+        TORCH_CUDA_ARCH_LIST: "8.0;8.6;8.9;9.0+PTX" # all "6.0;6.1;6.2;7.0;7.5;8.0;8.6;8.9;9.0"
         SETUPTOOLS_VERSION: "69.5.1"
         DCUTLASS_NVCC_ARCHS: "80;86;89;90;90a" # 90a for H100 GPU 89:GeForce RTX 4090 
         DEEPSPEED_VERSION: "master"