File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -282,7 +282,7 @@ def dockerfile_for_linux(output_file):
282
282
)
283
283
cuda_archs = "53;62;72;87"
284
284
else :
285
- cuda_archs = "60;61;70; 75;80;86;90"
285
+ cuda_archs = "75;80;86;90"
286
286
287
287
df += """
288
288
WORKDIR /workspace/onnxruntime
@@ -477,7 +477,7 @@ def dockerfile_for_windows(output_file):
477
477
WORKDIR /workspace/onnxruntime
478
478
ARG VS_DEVCMD_BAT="\BuildTools\VC\Auxiliary\Build\v cvars64.bat"
479
479
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 {}
481
481
""" .format (
482
482
ep_flags
483
483
)
You can’t perform that action at this time.
0 commit comments