TensorRT-LLMs/docker/common/install_cmake.sh
Martin Marciniszyn Mehringer d0e672f96d
chore: [TRTLLM-325][infra] Prepare for NGC release - reduce size of the docker images (#3990)
* chore: reduce size of the docker images

Signed-off-by: Martin Marciniszyn Mehringer <11665257+martinmarciniszyn@users.noreply.github.com>

* Finish the renaming script and run with new images.

Signed-off-by: Martin Marciniszyn Mehringer <11665257+MartinMarciniszyn@users.noreply.github.com>

* Fix installation of GCC toolset for Rocky Linux

Signed-off-by: Martin Marciniszyn Mehringer <11665257+MartinMarciniszyn@users.noreply.github.com>

* Upgrade to new docker images

Signed-off-by: Martin Marciniszyn Mehringer <11665257+MartinMarciniszyn@users.noreply.github.com>

---------

Signed-off-by: Martin Marciniszyn Mehringer <11665257+martinmarciniszyn@users.noreply.github.com>
Signed-off-by: Martin Marciniszyn Mehringer <11665257+MartinMarciniszyn@users.noreply.github.com>
2025-05-09 19:31:29 +08:00

29 lines
961 B
Bash

#!/bin/bash
set -ex
ARCH=$(uname -m)
CMAKE_VERSION="3.30.2"
GITHUB_URL="https://github.com"
if [ -n "${GITHUB_MIRROR}" ]; then
GITHUB_URL=${GITHUB_MIRROR}
fi
PARSED_CMAKE_VERSION=$(echo $CMAKE_VERSION | sed 's/\.[0-9]*$//')
CMAKE_FILE_NAME="cmake-${CMAKE_VERSION}-linux-${ARCH}"
RELEASE_URL_CMAKE=${GITHUB_URL}/Kitware/CMake/releases/download/v${CMAKE_VERSION}/${CMAKE_FILE_NAME}.tar.gz
wget --no-verbose ${RELEASE_URL_CMAKE} -P /tmp
tar -xf /tmp/${CMAKE_FILE_NAME}.tar.gz -C /usr/local/
ln -s /usr/local/${CMAKE_FILE_NAME} /usr/local/cmake
# Clean up temporary files
rm -rf /tmp/${CMAKE_FILE_NAME}.tar.gz
rm -rf /usr/local/${CMAKE_FILE_NAME}/doc
rm -rf /usr/local/${CMAKE_FILE_NAME}/man
rm -rf /usr/local/${CMAKE_FILE_NAME}/share/aclocal
rm -rf /usr/local/${CMAKE_FILE_NAME}/share/bash-completion
rm -rf /usr/local/${CMAKE_FILE_NAME}/share/emacs
rm -rf /usr/local/${CMAKE_FILE_NAME}/share/vim
echo 'export PATH=/usr/local/cmake/bin:$PATH' >> "${ENV}"