Skip to content

Commit d5fe74f

Browse files
authored
Remove unsupported compute capabilities (#275)
* Enable support for TensoRT 10.5 (#274) * Enable support for TensoRT 10.5 * Revert linux changes * Remove unsupported compute capabilities
1 parent 17f9596 commit d5fe74f

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,7 @@ def dockerfile_for_linux(output_file):
278278
)
279279
cuda_archs = "53;62;72;87"
280280
else:
281-
cuda_archs = "60;61;70;75;80;86;90"
281+
cuda_archs = "75;80;86;90"
282282

283283
df += """
284284
WORKDIR /workspace/onnxruntime
@@ -441,7 +441,9 @@ def dockerfile_for_windows(output_file):
441441
ARG ONNXRUNTIME_VERSION
442442
ARG ONNXRUNTIME_REPO
443443
RUN git clone -b rel-%ONNXRUNTIME_VERSION% --recursive %ONNXRUNTIME_REPO% onnxruntime && \
444-
(cd onnxruntime && git submodule update --init --recursive)
444+
cd onnxruntime && git submodule update --init --recursive && \
445+
git config --global user.email "you@example.com" && git config --global user.name "Your Name" && \
446+
git cherry-pick 709368ea1443dc1ff68dece31d692ad065fb94d4
445447
"""
446448

447449
if FLAGS.onnx_tensorrt_tag != "":
@@ -471,7 +473,7 @@ def dockerfile_for_windows(output_file):
471473
WORKDIR /workspace/onnxruntime
472474
ARG VS_DEVCMD_BAT="\BuildTools\VC\Auxiliary\Build\vcvars64.bat"
473475
RUN powershell Set-Content 'build.bat' -value 'call %VS_DEVCMD_BAT%',(Get-Content 'build.bat')
474-
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 {}
476+
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 {}
475477
""".format(
476478
ep_flags
477479
)

0 commit comments

Comments
 (0)