Apply UCX workaround (#6254) (#6307)

triton-inference-server · Sep 13, 2023 · 508dc44 · 508dc44
1 parent 27e5f0b
commit 508dc44
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/build.py b/build.py
@@ -1161,6 +1161,9 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
 LABEL com.nvidia.tritonserver.version="${TRITON_SERVER_VERSION}"
 
 ENV PATH /opt/tritonserver/bin:${PATH}
+# Remove once https://github.com/openucx/ucx/pull/9148 is available
+# in the min container.
+ENV UCX_MEM_EVENTS no
 """
 
     # TODO Remove once the ORT-OpenVINO "Exception while Reading network" is fixed