From 7b1a20d0b8138891ba48d66e5d08166311d7c2fc Mon Sep 17 00:00:00 2001
From: lingyuzeng <pylyzeng@gmail.com>
Date: Mon, 15 Jul 2024 12:03:45 +0800
Subject: [PATCH] add 4090 arch 8.9

---
 finetune/Dockerfile.update | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/finetune/Dockerfile.update b/finetune/Dockerfile.update
index d50cf7d..7687d05 100644
--- a/finetune/Dockerfile.update
+++ b/finetune/Dockerfile.update
@@ -111,7 +111,7 @@ python3 -m pip uninstall -y apex
 EOT
 
 # install apex
-ARG TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.6;9.0"
+ARG TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.6;8.9,9.0"
 ENV TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST}
 RUN <<EOT
 #!/bin/bash