Add script that correctly handles driver installation on Rocky Linux 8.

ci/build_conda.sh

@@ -10,10 +10,7 @@ export CMAKE_GENERATOR=Ninja
 rapids-print-env
 
 rapids-logger "Install CUDA Toolkit"
-yum update
-yum install -y epel-release
-yum install -y nvidia-driver-latest-dkms
-yum install -y cuda-toolkit-12-3
+source "$(dirname "$0")/install_latest_cuda_toolkit.sh"
 
 rapids-logger "Begin py build"
 

ci/build_wheel.sh

@@ -3,10 +3,7 @@
 
 
 rapids-logger "Install CUDA Toolkit"
-yum update
-yum install -y epel-release
-yum install -y nvidia-driver-latest-dkms
-yum install -y cuda-toolkit-12-3
+source "$(dirname "$0")/install_latest_cuda_toolkit.sh"
 
 rapids-logger "Check conda environment"
 conda list

ci/install_latest_cuda_toolkit.sh

@@ -0,0 +1,20 @@
+#!/bin/bash
+# Copyright (c) 2024, NVIDIA CORPORATION
+
+# Installs the latest CUDA Toolkit.
+# Supports CentOS 7 and Rocky Linux 8.
+
+yum update
+yum install -y epel-release
+
+OS_ID=$(. /etc/os-release; echo $ID)
+if [ "${OS_ID}" == "centos" ]; then
+    yum install -y nvidia-driver-latest-dkms
+elif [ "${OS_ID}" == "rocky" ]; then
+    yum install -y nvidia-driver
+else
+    echo "Error: OS not detected as CentOS or Rocky Linux. Exiting."
+    exit 1
+fi
+
+yum install -y cuda-toolkit-12-3