Reapply "Remove unsupported compute capabilities (#275)" (#281)

This reverts commit 2ddd44c.
triton-inference-server · Nov 1, 2024 · 56ce169 · 56ce169
1 parent 2828367
commit 56ce169
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py
@@ -282,7 +282,7 @@ def dockerfile_for_linux(output_file):
         )
         cuda_archs = "53;62;72;87"
     else:
-        cuda_archs = "60;61;70;75;80;86;90"
+        cuda_archs = "75;80;86;90"
 
     df += """
 WORKDIR /workspace/onnxruntime
@@ -477,7 +477,7 @@ def dockerfile_for_windows(output_file):
 WORKDIR /workspace/onnxruntime
 ARG VS_DEVCMD_BAT="\BuildTools\VC\Auxiliary\Build\vcvars64.bat"
 RUN powershell Set-Content 'build.bat' -value 'call %VS_DEVCMD_BAT%',(Get-Content 'build.bat')
-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 {}
+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 {}
 """.format(
         ep_flags
     )