Skip to content

Commit e32a32c

Browse files
authored
Reduce docker memory limit to WAR resource in use issue. Build ORT in parallel (#119)
1 parent c8041cf commit e32a32c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ else()
149149
endif()
150150

151151
set(TRITON_ONNXRUNTIME_DOCKER_IMAGE "tritonserver_onnxruntime")
152-
set(TRITON_ONNXRUNTIME_DOCKER_MEMORY "16g")
152+
set(TRITON_ONNXRUNTIME_DOCKER_MEMORY "8g")
153153
set(TRITON_ONNXRUNTIME_INCLUDE_PATHS "${CMAKE_CURRENT_BINARY_DIR}/onnxruntime/include")
154154
set(TRITON_ONNXRUNTIME_LIB_PATHS "${CMAKE_CURRENT_BINARY_DIR}/onnxruntime/lib")
155155
if (WIN32)

tools/gen_ort_dockerfile.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -386,7 +386,7 @@ def dockerfile_for_windows(output_file):
386386
WORKDIR /workspace/onnxruntime
387387
ARG VS_DEVCMD_BAT="\BuildTools\VC\Auxiliary\Build\vcvars64.bat"
388388
RUN powershell Set-Content 'build.bat' -value 'call %VS_DEVCMD_BAT%',(Get-Content 'build.bat')
389-
RUN build.bat --cmake_generator "Visual Studio 16 2019" --config Release --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=52;60;61;70;75;80;86" --skip_submodule_sync --build_shared_lib --update --build --build_dir /workspace/build {}
389+
RUN build.bat --cmake_generator "Visual Studio 16 2019" --config Release --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=52;60;61;70;75;80;86" --skip_submodule_sync --parallel --build_shared_lib --update --build --build_dir /workspace/build {}
390390
'''.format(ep_flags)
391391

392392
df += '''

0 commit comments

Comments
 (0)