File tree 1 file changed +5
-3
lines changed
1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -278,7 +278,7 @@ def dockerfile_for_linux(output_file):
278
278
)
279
279
cuda_archs = "53;62;72;87"
280
280
else :
281
- cuda_archs = "60;61;70; 75;80;86;90"
281
+ cuda_archs = "75;80;86;90"
282
282
283
283
df += """
284
284
WORKDIR /workspace/onnxruntime
@@ -441,7 +441,9 @@ def dockerfile_for_windows(output_file):
441
441
ARG ONNXRUNTIME_VERSION
442
442
ARG ONNXRUNTIME_REPO
443
443
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
445
447
"""
446
448
447
449
if FLAGS .onnx_tensorrt_tag != "" :
@@ -471,7 +473,7 @@ def dockerfile_for_windows(output_file):
471
473
WORKDIR /workspace/onnxruntime
472
474
ARG VS_DEVCMD_BAT="\BuildTools\VC\Auxiliary\Build\v cvars64.bat"
473
475
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 {}
475
477
""" .format (
476
478
ep_flags
477
479
)
You can’t perform that action at this time.
0 commit comments