File tree Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -90,6 +90,7 @@ FROM ${TRITON_IMAGE}:${TRITON_BASE_TAG} AS triton
9090
9191FROM devel AS tritondevel
9292
93+ ARG GITHUB_MIRROR=""
9394COPY --from=triton /opt/tritonserver/backends/python /opt/tritonserver/backends/python
9495COPY --from=triton /opt/tritonserver/lib /opt/tritonserver/lib
9596COPY --from=triton /opt/tritonserver/include /opt/tritonserver/include
@@ -100,7 +101,7 @@ COPY --from=triton /opt/tritonserver/caches /opt/tritonserver/caches
100101COPY 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
105106FROM ${DEVEL_IMAGE} AS wheel
106107WORKDIR /src/tensorrt_llm
Original file line number Diff line number Diff line change @@ -4,10 +4,16 @@ set -ex
44
55CUDA_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+
713install_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
You can’t perform that action at this time.
0 commit comments