From 9984765abb559648d6212996a3904db2b6bb4f77 Mon Sep 17 00:00:00 2001
From: Andy Georges <itkovian@gmail.com>
Date: Fri, 13 Dec 2024 21:39:30 +0100
Subject: [PATCH] fix: nvidia driver version on el9

---
 build.sh | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/build.sh b/build.sh
index 7b0c622085d..c5b9ea569f0 100755
--- a/build.sh
+++ b/build.sh
@@ -33,9 +33,10 @@ if grep "release 8.8" /etc/redhat-release; then
    CUDA_VERSION=${CUDA_VERSION:-12.6}
    CUDA_NVML_PKG="cuda-nvml-devel-${CUDA_VERSION//./-}"
 elif grep "release 9.4" /etc/redhat-release; then
-    NVDRV_NVML_PKG="libnvidia-ml"
-    CUDA_VERSION=${CUDA_VERSION:-12.6}
-    CUDA_NVML_PKG="cuda-nvml-devel-${CUDA_VERSION//./-}"
+   NVIDIA_DRIVER=${NVIDIA_DRIVER-555.42.06}
+   NVDRV_NVML_PKG="nvidia-driver-NVML${NVIDIA_DRIVER:+-$NVIDIA_DRIVER}"
+   CUDA_VERSION=${CUDA_VERSION:-12.6}
+   CUDA_NVML_PKG="cuda-nvml-devel-${CUDA_VERSION//./-}"
 fi
 
 # Prepare directory structure