File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 += """
284284WORKDIR /workspace/onnxruntime
@@ -473,7 +473,7 @@ def dockerfile_for_windows(output_file):
473473WORKDIR /workspace/onnxruntime
474474ARG VS_DEVCMD_BAT="\BuildTools\VC\Auxiliary\Build\v cvars64.bat"
475475RUN powershell Set-Content 'build.bat' -value 'call %VS_DEVCMD_BAT%',(Get-Content 'build.bat')
476- 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 {}
477477""" .format (
478478 ep_flags
479479 )
You can’t perform that action at this time.
0 commit comments