Skip to content

Commit e53be15

Browse files
authored
[TRTLLM-9213][infra] Fix boost issue (NVIDIA#8996)
Signed-off-by: ZhanruiSunCh <[email protected]>
1 parent c836ae5 commit e53be15

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

docker/Dockerfile.multi

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ FROM ${TRITON_IMAGE}:${TRITON_BASE_TAG} AS triton
9090

9191
FROM devel AS tritondevel
9292

93+
ARG GITHUB_MIRROR=""
9394
COPY --from=triton /opt/tritonserver/backends/python /opt/tritonserver/backends/python
9495
COPY --from=triton /opt/tritonserver/lib /opt/tritonserver/lib
9596
COPY --from=triton /opt/tritonserver/include /opt/tritonserver/include
@@ -100,7 +101,7 @@ COPY --from=triton /opt/tritonserver/caches /opt/tritonserver/caches
100101
COPY docker/common/install_triton.sh \
101102
./
102103

103-
RUN bash ./install_triton.sh && rm install_triton.sh
104+
RUN GITHUB_MIRROR=${GITHUB_MIRROR} bash ./install_triton.sh && rm install_triton.sh
104105

105106
FROM ${DEVEL_IMAGE} AS wheel
106107
WORKDIR /src/tensorrt_llm

docker/common/install_triton.sh

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,16 @@ set -ex
44

55
CUDA_VER="13"
66

7+
if [ -n "${GITHUB_MIRROR}" ]; then
8+
BOOST_URL="https://urm.nvidia.com/artifactory/sw-dl-triton-generic-local/triton/ci-cd/binaries/boost/1.80.0/boost_1_80_0.tar.gz"
9+
else
10+
BOOST_URL="https://archives.boost.io/release/1.80.0/source/boost_1_80_0.tar.gz"
11+
fi
12+
713
install_boost() {
814
# Install boost version >= 1.78 for boost::span
915
# Current libboost-dev apt packages are < 1.78, so install from tar.gz
10-
wget --retry-connrefused --timeout=180 --tries=10 --continue -O /tmp/boost.tar.gz https://archives.boost.io/release/1.80.0/source/boost_1_80_0.tar.gz \
16+
wget --no-verbose --retry-connrefused --timeout=180 --tries=10 --continue -O /tmp/boost.tar.gz ${BOOST_URL} \
1117
&& tar xzf /tmp/boost.tar.gz -C /tmp \
1218
&& mv /tmp/boost_1_80_0/boost /usr/include/boost \
1319
&& rm -rf /tmp/boost_1_80_0 /tmp/boost.tar.gz

0 commit comments

Comments
 (0)