Skip to content

Commit afc908a

Browse files
committed
Enable support for TensoRT 10.5
1 parent 17f9596 commit afc908a

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,9 @@ def dockerfile_for_linux(output_file):
237237
ARG ONNXRUNTIME_BUILD_CONFIG
238238
239239
RUN git clone -b rel-${ONNXRUNTIME_VERSION} --recursive ${ONNXRUNTIME_REPO} onnxruntime && \
240-
(cd onnxruntime && git submodule update --init --recursive)
240+
cd onnxruntime && git submodule update --init --recursive && \
241+
git config --global user.email "[email protected]" && git config --global user.name "Your Name" && \
242+
git cherry-pick 709368ea1443dc1ff68dece31d692ad065fb94d4
241243
"""
242244

243245
if FLAGS.onnx_tensorrt_tag != "":
@@ -441,7 +443,9 @@ def dockerfile_for_windows(output_file):
441443
ARG ONNXRUNTIME_VERSION
442444
ARG ONNXRUNTIME_REPO
443445
RUN git clone -b rel-%ONNXRUNTIME_VERSION% --recursive %ONNXRUNTIME_REPO% onnxruntime && \
444-
(cd onnxruntime && git submodule update --init --recursive)
446+
cd onnxruntime && git submodule update --init --recursive && \
447+
git config --global user.email "[email protected]" && git config --global user.name "Your Name" && \
448+
git cherry-pick 709368ea1443dc1ff68dece31d692ad065fb94d4
445449
"""
446450

447451
if FLAGS.onnx_tensorrt_tag != "":

0 commit comments

Comments
 (0)