Preload: libnvshmem_host.so

Author: mc-nv

dockerfile/Dockerfile.triton.trt_llm_backend

@@ -64,6 +64,8 @@ RUN [ "$(uname -m)" != "x86_64" ] && arch="sbsa" || arch="x86_64" \
         cuda-nvrtc-dev-${NVRTC_CUDA_VERSION}=${NVRTC_VER} \
         libnccl2=${NCCL_VER} \
         libnccl-dev=${NCCL_VER} \
+    && dpkg -L libnvshmem3-cuda-13 | grep libnvshmem_host.so | sed -e 's/libnvshmem_host.*//g' | sort -u > /etc/ld.so.conf.d/libnvshmem3-cuda-13.conf \
+    && ldconfig \
     && rm -rf /var/lib/apt/lists/*
 
 # Download & install TRT release