From 12fa109910b4433af0643bdbdb8ff6d4809fd66d Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Sun, 1 Jun 2025 13:37:11 -0600
Subject: [PATCH] Updated cuda arch list on docker build

---
 docker/Dockerfile | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 831e9e1..3093b06 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -5,6 +5,9 @@ LABEL authors="jaret"
 # Set noninteractive to avoid timezone prompts
 ENV DEBIAN_FRONTEND=noninteractive
 
+# ref https://en.wikipedia.org/wiki/CUDA
+ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0 10.0 12.0"
+
 # Install dependencies
 RUN apt-get update && apt-get install --no-install-recommends -y \
     git \