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 d666e38 commit 94bf57aCopy full SHA for 94bf57a
tools/gen_ort_dockerfile.py
@@ -324,7 +324,7 @@ def dockerfile_for_linux(output_file):
324
cuda_archs = "87"
325
else:
326
if os.uname().machine != "x86_64":
327
- cuda_archs = "all-major"
+ 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 = (
0 commit comments