File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -64,8 +64,6 @@ RUN [ "$(uname -m)" != "x86_64" ] && arch="sbsa" || arch="x86_64" \
6464 cuda-nvrtc-dev-${NVRTC_CUDA_VERSION}=${NVRTC_VER} \
6565 libnccl2=${NCCL_VER} \
6666 libnccl-dev=${NCCL_VER} \
67- && 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 \
68- && ldconfig \
6967 && rm -rf /var/lib/apt/lists/*
7068
7169# Download & install TRT release
@@ -163,6 +161,8 @@ RUN [ "$(uname -m)" != "x86_64" ] && arch="sbsa" || arch="x86_64" \
163161 libnccl2=${NCCL_VER} \
164162 libnccl-dev=${NCCL_VER} \
165163 libnvshmem3-cuda-13 \
164+ && 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 \
165+ && ldconfig \
166166 && rm -rf /var/lib/apt/lists/*
167167
168168# Install TRT
You can’t perform that action at this time.
0 commit comments