mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
fda8b0277a
* [TRTLLM-4374] Upgrade TRT 10.10.0 GA, CUDA 12.9 GA and DLFW 25.04 Signed-off-by: Yiqing Yan <yiqingy@nvidia.com> * fix review Signed-off-by: Yiqing Yan <yiqingy@nvidia.com> * update images Signed-off-by: Yiqing Yan <yiqingy@nvidia.com> * Update jenkins/L0_Test.groovy Co-authored-by: Yanchao Lu <yanchaol@nvidia.com> Signed-off-by: Yiqing Yan <yiqingy@nvidia.com> * update image name Signed-off-by: Yiqing Yan <yiqingy@nvidia.com> --------- Signed-off-by: Yiqing Yan <yiqingy@nvidia.com> Co-authored-by: Yanchao Lu <yanchaol@nvidia.com>
35 lines
1.2 KiB
Bash
35 lines
1.2 KiB
Bash
#!/bin/bash
|
|
|
|
set -ex
|
|
|
|
# This script is used for reinstalling CUDA on Rocky Linux 8 with the run file.
|
|
# CUDA version is usually aligned with the latest NGC CUDA image tag.
|
|
# Only use when public CUDA image is not ready.
|
|
CUDA_VER="12.9.0_575.51.03"
|
|
CUDA_VER_SHORT="${CUDA_VER%_*}"
|
|
|
|
NVCC_VERSION_OUTPUT=$(nvcc --version)
|
|
OLD_CUDA_VER=$(echo $NVCC_VERSION_OUTPUT | grep -oP "\d+\.\d+" | head -n 1)
|
|
echo "The version of pre-installed CUDA is ${OLD_CUDA_VER}."
|
|
|
|
reinstall_rockylinux_cuda() {
|
|
dnf -y install epel-release
|
|
dnf remove -y "cuda*" "*cublas*" "*cufft*" "*cufile*" "*curand*" "*cusolver*" "*cusparse*" "*gds-tools*" "*npp*" "*nvjpeg*" "nsight*" "*nvvm*"
|
|
rm -rf /usr/local/cuda-${OLD_CUDA_VER}
|
|
wget -q https://developer.download.nvidia.com/compute/cuda/${CUDA_VER_SHORT}/local_installers/cuda_${CUDA_VER}_linux.run
|
|
sh cuda_${CUDA_VER}_linux.run --silent --override --toolkit
|
|
rm -f cuda_${CUDA_VER}_linux.run
|
|
}
|
|
|
|
# Install base packages depending on the base OS
|
|
ID=$(grep -oP '(?<=^ID=).+' /etc/os-release | tr -d '"')
|
|
case "$ID" in
|
|
rocky)
|
|
echo "Reinstall CUDA for RockyLinux 8..."
|
|
reinstall_rockylinux_cuda
|
|
;;
|
|
*)
|
|
echo "Skip for other OS..."
|
|
;;
|
|
esac
|