Skip to content

Commit 72ca414

Browse files
committed
Moving changes to the right place in Dockerfile.
1 parent 64ae321 commit 72ca414

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dockerfile/Dockerfile.triton.trt_llm_backend

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)