diff --git a/CMakeLists.txt b/CMakeLists.txt index 9596d56..b15fe7b 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -157,7 +157,7 @@ else() endif() set(TRITON_ONNXRUNTIME_DOCKER_IMAGE "tritonserver_onnxruntime") - set(TRITON_ONNXRUNTIME_DOCKER_MEMORY "8g") + set(TRITON_ONNXRUNTIME_DOCKER_MEMORY "$,32g,8g>") set(TRITON_ONNXRUNTIME_INCLUDE_PATHS "${CMAKE_CURRENT_BINARY_DIR}/onnxruntime/include") set(TRITON_ONNXRUNTIME_LIB_PATHS "${CMAKE_CURRENT_BINARY_DIR}/onnxruntime/lib") if (WIN32)