Skip to content

Commit f03aeb6

Browse files
committed
Update CUDA archs in ORT
1 parent d646603 commit f03aeb6

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
@@ -283,7 +283,7 @@ def dockerfile_for_linux(output_file):
283283
)
284284
cuda_archs = "53;62;72;87"
285285
else:
286-
cuda_archs = "75;80;86;90"
286+
cuda_archs = "75;80;86;90;100;120"
287287

288288
df += """
289289
WORKDIR /workspace/onnxruntime
@@ -478,7 +478,7 @@ def dockerfile_for_windows(output_file):
478478
WORKDIR /workspace/onnxruntime
479479
ARG VS_DEVCMD_BAT="\BuildTools\VC\Auxiliary\Build\vcvars64.bat"
480480
RUN powershell Set-Content 'build.bat' -value 'call %VS_DEVCMD_BAT%',(Get-Content 'build.bat')
481-
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 {}
481+
RUN build.bat --cmake_generator "Visual Studio 17 2022" --config Release --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=75;80;86;90,100,120" --skip_submodule_sync --parallel --build_shared_lib --compile_no_warning_as_error --skip_tests --update --build --build_dir /workspace/build {}
482482
""".format(
483483
ep_flags
484484
)

0 commit comments

Comments
 (0)