File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -157,7 +157,7 @@ else()
157
157
endif ()
158
158
159
159
set (TRITON_ONNXRUNTIME_DOCKER_IMAGE "tritonserver_onnxruntime" )
160
- set (TRITON_ONNXRUNTIME_DOCKER_MEMORY "8g " )
160
+ set (TRITON_ONNXRUNTIME_DOCKER_MEMORY "$<IF:$<BOOL:WIN32>,32g,8g> " )
161
161
set (TRITON_ONNXRUNTIME_INCLUDE_PATHS "${CMAKE_CURRENT_BINARY_DIR} /onnxruntime/include" )
162
162
set (TRITON_ONNXRUNTIME_LIB_PATHS "${CMAKE_CURRENT_BINARY_DIR} /onnxruntime/lib" )
163
163
if (WIN32 )
Original file line number Diff line number Diff line change @@ -383,7 +383,8 @@ def dockerfile_for_linux(output_file):
383
383
# Linking compiled ONNX Runtime libraries to their corresponding versioned libraries
384
384
df += """
385
385
RUN cd /opt/onnxruntime/lib \
386
- && ln -s libonnxruntime.so libonnxruntime.so.${ONNXRUNTIME_VERSION}
386
+ && ln -s libonnxruntime.so libonnxruntime.so.1 \
387
+ && ln -s libonnxruntime.so.1 libonnxruntime.so.${ONNXRUNTIME_VERSION}
387
388
"""
388
389
df += """
389
390
RUN cd /opt/onnxruntime/lib && \
You can’t perform that action at this time.
0 commit comments