From 4a8611ad666ec621855afb89a570b1061bfd0cfd Mon Sep 17 00:00:00 2001
From: nico <nico@holaf.fr>
Date: Sun, 27 Jul 2025 15:00:30 +0200
Subject: [PATCH] ARCH 12 test

---
 Dockerfile           | 2 +-
 Dockerfile.buildbase | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 4433a50..01175ac 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,7 +14,7 @@ ENV BASE_DIR=/config \
 # Set compiler and Torch/CUDA architecture for any potential runtime compilations
 ENV CC=/usr/bin/gcc-13
 ENV CXX=/usr/bin/g++-13
-ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0 9.0a 10"
+ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0 9.0a 10 12"
 
 # --- System & Package Installation ---
 RUN apt-get update -q && \
diff --git a/Dockerfile.buildbase b/Dockerfile.buildbase
index 4d9929c..29d1752 100644
--- a/Dockerfile.buildbase
+++ b/Dockerfile.buildbase
@@ -3,7 +3,7 @@ FROM nvidia/cuda:12.8.1-cudnn-devel-ubuntu24.04 AS builder
 
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PIP_BREAK_SYSTEM_PACKAGES=1
-ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0 9.0a 10"
+ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0 9.0a 10 12"
 ENV MAX_JOBS=8
 WORKDIR /build