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")