Updated cuda arch list on docker build

2026-02-06 05:29:57 +00:00 · 2025-06-01 13:37:11 -06:00
parent 7d76165dcf
commit 12fa109910
1 changed files with 3 additions and 0 deletions
--- 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 \