From 17d7cce1c06e5904e322101beb87add67b9a13a7 Mon Sep 17 00:00:00 2001 From: Misha Chornyi <99709299+mc-nv@users.noreply.github.com> Date: Mon, 14 Oct 2024 11:00:15 -0700 Subject: [PATCH 1/2] Enable support for TensoRT 10.5 (#274) * Enable support for TensoRT 10.5 * Revert linux changes --- tools/gen_ort_dockerfile.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py index a9cd7b6..c320f60 100755 --- a/tools/gen_ort_dockerfile.py +++ b/tools/gen_ort_dockerfile.py @@ -441,7 +441,9 @@ 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) + 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 """ if FLAGS.onnx_tensorrt_tag != "": From d1bb37ba48b947b17ab478d69160a0994b4682b9 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Tue, 15 Oct 2024 09:22:23 -0700 Subject: [PATCH 2/2] Remove unsupported compute capabilities --- 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 c320f60..436b2f2 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 = "60;61;70;75;80;86;90" + cuda_archs = "75;80;86;90" df += """ WORKDIR /workspace/onnxruntime @@ -473,7 +473,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 )