diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py index 2e8271e..efae8cd 100755 --- 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 )