mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-22 19:52:38 +08:00
8124a62b74
Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com> Signed-off-by: Zhanrui Sun <184402041+ZhanruiSunCh@users.noreply.github.com> Signed-off-by: Yanchao Lu <yanchaol@nvidia.com> Co-authored-by: Yanchao Lu <yanchaol@nvidia.com>
33 lines
935 B
Bash
33 lines
935 B
Bash
#!/bin/bash
|
|
set -ex
|
|
|
|
UCX_VERSION="v1.19.x"
|
|
UCX_INSTALL_PATH="/usr/local/ucx/"
|
|
CUDA_PATH="/usr/local/cuda"
|
|
UCX_REPO="https://github.com/openucx/ucx.git"
|
|
|
|
if [ -n "${GITHUB_MIRROR}" ]; then
|
|
export UCX_REPO="https://gitlab-master.nvidia.com/ftp/GitHubSync/ucx.git"
|
|
fi
|
|
|
|
rm -rf ${UCX_INSTALL_PATH}
|
|
git clone --depth 1 -b ${UCX_VERSION} ${UCX_REPO}
|
|
cd ucx
|
|
./autogen.sh
|
|
./contrib/configure-release \
|
|
--prefix=${UCX_INSTALL_PATH} \
|
|
--enable-shared \
|
|
--disable-static \
|
|
--disable-doxygen-doc \
|
|
--enable-optimizations \
|
|
--enable-cma \
|
|
--enable-devel-headers \
|
|
--with-cuda=${CUDA_PATH} \
|
|
--with-verbs \
|
|
--with-dm \
|
|
--enable-mt
|
|
make install -j$(nproc)
|
|
cd ..
|
|
rm -rf ucx # Remove UCX source to save space
|
|
echo "export LD_LIBRARY_PATH=${UCX_INSTALL_PATH}/lib:\$LD_LIBRARY_PATH" >> "${ENV}"
|