Add libnvshmem installation

build.py

@@ -1318,8 +1318,13 @@ 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() not in ["igpu", "windows", "rhel"]:
+            backend_dependencies += " libnvshmem3-cuda-13"
+
     # openssh-server is needed for fastertransformer
     if "fastertransformer" in backends:
         backend_dependencies += " openssh-server"