build.py - add shared-memory instalation back to the ManyLinux

Author: mc-nv

build.py

@@ -1520,9 +1520,16 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
 """
 
     if target_platform() == "rhel":
+        repo_arch = "sbsa" if target_machine == "aarch64" else "x86_64"
         df += """
-RUN dirname  $(find /usr -name "libcudart*.so" -o  -name "libnvinf*.so" -o -name "libnvshm*" ) | sort -u > /etc/ld.so.conf.d/triton-cuda-libs.conf && ldconfig
-"""
+RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/{repo_arch}/cuda-rhel8.repo \\
+    && dnf clean expire-cache \\
+    && dnf install --assumeyes libnvshmem3-cuda-13
+
+RUN dirname  $(find /usr -name "libcudart*.so" -o  -name "libnvinf*.so" -o -name "libnvshm*" -type f) | sort -u > /etc/ld.so.conf.d/triton-cuda-libs.conf && ldconfig
+""".format(
+            repo_arch=repo_arch
+        )
 
     df += """
 WORKDIR /opt/tritonserver