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