From ec7c2cde6f920761598e148bbb156de3add1f559 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Fri, 1 Nov 2024 11:55:45 -0700 Subject: [PATCH] Reapply "Remove unsupported compute capabilities (#275)" This reverts commit 2ddd44cea4fbeae025fe643685abdb3b26d68ca4. --- tools/gen_ort_dockerfile.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py index 2e8271e..efae8cd 100755 --- a/tools/gen_ort_dockerfile.py +++ b/tools/gen_ort_dockerfile.py @@ -282,7 +282,7 @@ def dockerfile_for_linux(output_file): ) cuda_archs = "53;62;72;87" else: - cuda_archs = "60;61;70;75;80;86;90" + cuda_archs = "75;80;86;90" df += """ WORKDIR /workspace/onnxruntime @@ -477,7 +477,7 @@ def dockerfile_for_windows(output_file): WORKDIR /workspace/onnxruntime ARG VS_DEVCMD_BAT="\BuildTools\VC\Auxiliary\Build\vcvars64.bat" RUN powershell Set-Content 'build.bat' -value 'call %VS_DEVCMD_BAT%',(Get-Content 'build.bat') -RUN build.bat --cmake_generator "Visual Studio 17 2022" --config Release --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=60;61;70;75;80;86;90" --skip_submodule_sync --parallel --build_shared_lib --compile_no_warning_as_error --skip_tests --update --build --build_dir /workspace/build {} +RUN build.bat --cmake_generator "Visual Studio 17 2022" --config Release --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=75;80;86;90" --skip_submodule_sync --parallel --build_shared_lib --compile_no_warning_as_error --skip_tests --update --build --build_dir /workspace/build {} """.format( ep_flags )