Fix issue with building gpu drivers on rl8

GoogleCloudPlatform · Dec 13, 2023 · 0e5f4bb · 0e5f4bb
1 parent 3706450
commit 0e5f4bb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/integration_test/third_party_apps_data/applications/dcgm/centos_rhel/install b/integration_test/third_party_apps_data/applications/dcgm/centos_rhel/install
@@ -31,7 +31,7 @@ case $DEVICE_CODE in
         ;;
     *)
         # Installing latest version of NVIDIA CUDA and driver
-        DRIVER_VERSION=535.104.05
+        DRIVER_VERSION=535.129.03
         ;;
 esac
 

diff --git a/integration_test/third_party_apps_data/applications/nvml/centos_rhel/install b/integration_test/third_party_apps_data/applications/nvml/centos_rhel/install
@@ -32,16 +32,17 @@ case $DEVICE_CODE in
         ;;
     *)
         # Installing latest version of NVIDIA CUDA and driver
-        DRIVER_VERSION=535.104.05
+        DRIVER_VERSION=535.129.03
         CUDA_VERSION=12.2.2
+        CUDA_BUNDLED_DRIVER_VERSION=535.104.05
         ;;
 esac
 
 echo "Installing NVIDIA CUDA $CUDA_VERSION with driver $DRIVER_VERSION"
 curl -fSsl -O https://us.download.nvidia.com/tesla/$DRIVER_VERSION/NVIDIA-Linux-x86_64-$DRIVER_VERSION.run
 sudo bash ./NVIDIA-Linux-x86_64-$DRIVER_VERSION.run --silent
-wget --no-verbose https://developer.download.nvidia.com/compute/cuda/$CUDA_VERSION/local_installers/cuda_${CUDA_VERSION}_${DRIVER_VERSION}_linux.run
-sudo sh cuda_${CUDA_VERSION}_${DRIVER_VERSION}_linux.run --toolkit --silent
+curl -fSsl -O https://developer.download.nvidia.com/compute/cuda/$CUDA_VERSION/local_installers/cuda_${CUDA_VERSION}_${CUDA_BUNDLED_DRIVER_VERSION}_linux.run
+sudo sh cuda_${CUDA_VERSION}_${CUDA_BUNDLED_DRIVER_VERSION}_linux.run --toolkit --silent
 
 # check NVIDIA driver installation succeeded
 nvidia-smi
-Original file line number
+Diff line change
@@ Expand Up / @@ -31,7 +31,7 @@ case $DEVICE_CODE in @@
             ;;
         *)
             # Installing latest version of NVIDIA CUDA and driver
-            DRIVER_VERSION=535.104.05
+            DRIVER_VERSION=535.129.03
             ;;
     esac
@@ Expand Down @@