From 2594aa37ef73526df34c939ef2ef7da2b32c3f50 Mon Sep 17 00:00:00 2001 From: Misha Chornyi <99709299+mc-nv@users.noreply.github.com> Date: Fri, 26 Sep 2025 22:51:40 -0700 Subject: [PATCH] Define missed CUDA architectures for non x86_64 builds. (#318) * Enable all-major CUDA architectures for non x86_64 builds. * Hardcoding the value --- tools/gen_ort_dockerfile.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py index f535229..2b56564 100755 --- a/tools/gen_ort_dockerfile.py +++ b/tools/gen_ort_dockerfile.py @@ -323,7 +323,9 @@ def dockerfile_for_linux(output_file): else: cuda_archs = "87" else: - if os.getenv("CUDA_ARCH_LIST") is not None: + if os.uname().machine != "x86_64": + cuda_archs = "80;86;90;100;110;120;121" + elif os.getenv("CUDA_ARCH_LIST") is not None: print(f"[INFO] Defined CUDA_ARCH_LIST: {os.getenv('CUDA_ARCH_LIST')}") cuda_archs = ( os.getenv("CUDA_ARCH_LIST")