Skip to content

Commit cd5cd60

Browse files
[None][infra] Move install_boost from install_triton.sh to install_base.sh (#10055)
Signed-off-by: Iman Tabrizian <[email protected]> Signed-off-by: Zhanrui Sun <[email protected]> Co-authored-by: Zhanrui Sun <[email protected]>
1 parent 8462cf6 commit cd5cd60

File tree

3 files changed

+21
-20
lines changed

3 files changed

+21
-20
lines changed

docker/common/install_base.sh

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,21 @@ if [ -n "${GITHUB_MIRROR}" ]; then
88
export PIP_INDEX_URL="https://urm.nvidia.com/artifactory/api/pypi/pypi-remote/simple"
99
fi
1010

11+
if [ -n "${GITHUB_MIRROR}" ]; then
12+
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"
13+
else
14+
BOOST_URL="https://archives.boost.io/release/1.80.0/source/boost_1_80_0.tar.gz"
15+
fi
16+
17+
install_boost() {
18+
# Install boost version >= 1.78 for boost::span
19+
# Current libboost-dev apt packages are < 1.78, so install from tar.gz
20+
wget --no-verbose --retry-connrefused --timeout=180 --tries=10 --continue -O /tmp/boost.tar.gz ${BOOST_URL} \
21+
&& tar xzf /tmp/boost.tar.gz -C /tmp \
22+
&& mv /tmp/boost_1_80_0/boost /usr/include/boost \
23+
&& rm -rf /tmp/boost_1_80_0 /tmp/boost.tar.gz
24+
}
25+
1126
set_bash_env() {
1227
if [ ! -f ${BASH_ENV} ];then
1328
touch ${BASH_ENV}
@@ -151,11 +166,13 @@ set_bash_env
151166
case "$ID" in
152167
ubuntu)
153168
init_ubuntu
169+
install_boost
154170
;;
155171
rocky)
156172
install_python_rockylinux $1
157173
install_pyp_rockylinux
158174
install_gcctoolset_rockylinux
175+
install_boost
159176
;;
160177
*)
161178
echo "Unable to determine OS..."

docker/common/install_triton.sh

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -4,21 +4,6 @@ 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-
13-
install_boost() {
14-
# Install boost version >= 1.78 for boost::span
15-
# Current libboost-dev apt packages are < 1.78, so install from tar.gz
16-
wget --no-verbose --retry-connrefused --timeout=180 --tries=10 --continue -O /tmp/boost.tar.gz ${BOOST_URL} \
17-
&& tar xzf /tmp/boost.tar.gz -C /tmp \
18-
&& mv /tmp/boost_1_80_0/boost /usr/include/boost \
19-
&& rm -rf /tmp/boost_1_80_0 /tmp/boost.tar.gz
20-
}
21-
227
install_triton_deps() {
238
apt-get update \
249
&& apt-get install -y --no-install-recommends \
@@ -30,7 +15,6 @@ install_triton_deps() {
3015
libb64-dev \
3116
libarchive-dev \
3217
datacenter-gpu-manager-4-cuda${CUDA_VER} \
33-
&& install_boost \
3418
&& apt-get clean \
3519
&& rm -rf /var/lib/apt/lists/*
3620
}

jenkins/current_image_tags.properties

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# images are adopted from PostMerge pipelines, the abbreviated commit hash is used instead.
1414
IMAGE_NAME=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm
1515

16-
LLM_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.10-py3-x86_64-ubuntu24.04-trt10.13.3.9-skip-tritondevel-202512151112-9977
17-
LLM_SBSA_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.10-py3-aarch64-ubuntu24.04-trt10.13.3.9-skip-tritondevel-202512151112-9977
18-
LLM_ROCKYLINUX8_PY310_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:cuda-13.0.2-devel-rocky8-x86_64-rocky8-py310-trt10.13.3.9-skip-tritondevel-202512151112-9977
19-
LLM_ROCKYLINUX8_PY312_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:cuda-13.0.2-devel-rocky8-x86_64-rocky8-py312-trt10.13.3.9-skip-tritondevel-202512151112-9977
16+
LLM_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.10-py3-x86_64-ubuntu24.04-trt10.13.3.9-skip-tritondevel-202512241744-10055
17+
LLM_SBSA_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.10-py3-aarch64-ubuntu24.04-trt10.13.3.9-skip-tritondevel-202512241744-10055
18+
LLM_ROCKYLINUX8_PY310_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:cuda-13.0.2-devel-rocky8-x86_64-rocky8-py310-trt10.13.3.9-skip-tritondevel-202512241744-10055
19+
LLM_ROCKYLINUX8_PY312_DOCKER_IMAGE=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:cuda-13.0.2-devel-rocky8-x86_64-rocky8-py312-trt10.13.3.9-skip-tritondevel-202512241744-10055

0 commit comments

Comments
 (0)