We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e0c13ae commit c8e73d3Copy full SHA for c8e73d3
tools/gen_ort_dockerfile.py
@@ -323,7 +323,9 @@ def dockerfile_for_linux(output_file):
323
else:
324
cuda_archs = "87"
325
326
- if os.getenv("CUDA_ARCH_LIST") is not None:
+ if os.uname().machine != "x86_64":
327
+ cuda_archs = "80;86;90;100;110;120;121"
328
+ elif os.getenv("CUDA_ARCH_LIST") is not None:
329
print(f"[INFO] Defined CUDA_ARCH_LIST: {os.getenv('CUDA_ARCH_LIST')}")
330
cuda_archs = (
331
os.getenv("CUDA_ARCH_LIST")
0 commit comments