Add libnvshmem installation

build.py

@@ -1318,8 +1318,17 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
         backend_dependencies = "libgomp1"
 
     # libgfortran5 is needed by pytorch backend on ARM
-    if ("pytorch" in backends) and (target_machine == "aarch64"):
-        backend_dependencies += " libgfortran5"
+    if "pytorch" in backends:
+        if target_machine == "aarch64":
+            backend_dependencies += " libgfortran5"
+
+        if (
+            target_platform != "igpu"
+            or target_platform != "windows"
+            or target_platform != "rhel"
+        ):
+            backend_dependencies += " libnvshmem3-cuda-13"
+
     # openssh-server is needed for fastertransformer
     if "fastertransformer" in backends:
         backend_dependencies += " openssh-server"