Apply UCX workaround (#6254)

triton-inference-server · Sep 11, 2023 · e0dc4aa · e0dc4aa
1 parent 9fe213a
commit e0dc4aa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/build.py b/build.py
@@ -1149,6 +1149,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