From 2ddd44cea4fbeae025fe643685abdb3b26d68ca4 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Fri, 18 Oct 2024 17:05:54 -0700 Subject: [PATCH] Revert "Remove unsupported compute capabilities (#275)" This reverts commit d5fe74f73109425a09ed3332029d270e11da5311. --- tools/gen_ort_dockerfile.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py index 436b2f2..a9cd7b6 100755 --- a/tools/gen_ort_dockerfile.py +++ b/tools/gen_ort_dockerfile.py @@ -278,7 +278,7 @@ def dockerfile_for_linux(output_file): ) cuda_archs = "53;62;72;87" else: - cuda_archs = "75;80;86;90" + cuda_archs = "60;61;70;75;80;86;90" df += """ WORKDIR /workspace/onnxruntime @@ -441,9 +441,7 @@ def dockerfile_for_windows(output_file): ARG ONNXRUNTIME_VERSION ARG ONNXRUNTIME_REPO RUN git clone -b rel-%ONNXRUNTIME_VERSION% --recursive %ONNXRUNTIME_REPO% onnxruntime && \ - cd onnxruntime && git submodule update --init --recursive && \ - git config --global user.email "you@example.com" && git config --global user.name "Your Name" && \ - git cherry-pick 709368ea1443dc1ff68dece31d692ad065fb94d4 + (cd onnxruntime && git submodule update --init --recursive) """ if FLAGS.onnx_tensorrt_tag != "": @@ -473,7 +471,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=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=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 {} """.format( ep_flags )