Skip to content

Commit 56ce169

Browse files
authored
Reapply "Remove unsupported compute capabilities (#275)" (#281)
This reverts commit 2ddd44c.
1 parent 2828367 commit 56ce169

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,7 @@ def dockerfile_for_linux(output_file):
282282
)
283283
cuda_archs = "53;62;72;87"
284284
else:
285-
cuda_archs = "60;61;70;75;80;86;90"
285+
cuda_archs = "75;80;86;90"
286286

287287
df += """
288288
WORKDIR /workspace/onnxruntime
@@ -477,7 +477,7 @@ def dockerfile_for_windows(output_file):
477477
WORKDIR /workspace/onnxruntime
478478
ARG VS_DEVCMD_BAT="\BuildTools\VC\Auxiliary\Build\vcvars64.bat"
479479
RUN powershell Set-Content 'build.bat' -value 'call %VS_DEVCMD_BAT%',(Get-Content 'build.bat')
480-
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 {}
480+
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 {}
481481
""".format(
482482
ep_flags
483483
)

0 commit comments

Comments
 (0)