mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-13 22:18:36 +08:00
[None][infra] Update ucx to 1.20.x (#9977)
Signed-off-by: Yoray Zack <yorayz@nvidia.com> Signed-off-by: Yoray Zack <62789610+zackyoray@users.noreply.github.com>
This commit is contained in:
parent
4f75a31a45
commit
63e7a2fa70
@ -1,7 +1,8 @@
|
||||
#!/bin/bash
|
||||
set -ex
|
||||
|
||||
UCX_VERSION="v1.19.x"
|
||||
UCX_VERSION="v1.20.x"
|
||||
UCX_COMMIT="f656dbdf93e72e60b5d6ca78b9e3d9e744e789bd"
|
||||
UCX_INSTALL_PATH="/usr/local/ucx/"
|
||||
CUDA_PATH="/usr/local/cuda"
|
||||
UCX_REPO="https://github.com/openucx/ucx.git"
|
||||
@ -9,7 +10,10 @@ UCX_REPO="https://github.com/openucx/ucx.git"
|
||||
mkdir -p /third-party-source
|
||||
|
||||
rm -rf ${UCX_INSTALL_PATH}
|
||||
git clone --depth 1 -b ${UCX_VERSION} ${UCX_REPO}
|
||||
git clone -b ${UCX_VERSION} ${UCX_REPO}
|
||||
cd ucx
|
||||
git checkout ${UCX_COMMIT}
|
||||
cd ..
|
||||
tar -czf /third-party-source/ucx-${UCX_VERSION}.tar.gz ucx
|
||||
cd ucx
|
||||
./autogen.sh
|
||||
|
||||
@ -13,7 +13,7 @@
|
||||
# images are adopted from PostMerge pipelines, the abbreviated commit hash is used instead.
|
||||
IMAGE_NAME=urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm
|
||||
|
||||
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-202512121105-9707
|
||||
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-202512121105-9707
|
||||
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-202512121105-9707
|
||||
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-202512121105-9707
|
||||
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
|
||||
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
|
||||
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
|
||||
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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user