Merge ~dannf/+git/maas-preseeds:oot-driver-updates into ~ce-hyperscale/+git/maas-preseeds:main

[dann frazier]

[Hon Ming Hui (hm-hui)]

+1

review: Approve

diff --git a/late.sh b/late.sh
index 29cfe7e..0b260b3 100755
--- a/late.sh
+++ b/late.sh
@@ -4,18 +4,25 @@ set -e
 set -x
 
 export DEBIAN_FRONTEND=noninteractive
-NVIDIA_DRIVER_VERSION=470-server
+UBUNTU_VER="$(lsb_release -rs)"
+case $UBUNTU_VER in
+    18.04|20.04)
+	NVIDIA_DRIVER_VERSION=470-server
+	;;
+    *)
+	NVIDIA_DRIVER_VERSION=525-server
+	;;
+esac
+	
 
 install_mellanox_ofed() {
     local MLNX_REPO="https://linux.mellanox.com/public/repo/mlnx_ofed"
-    local ubuntu_ver
-    ubuntu_ver="$(lsb_release -rs)"
-    case $ubuntu_ver in
+    case $UBUNTU_VER in
         20.04)
             MELLANOX_OFED_VERSION=5.4-3.5.8.0
             ;;
         18.04)
-            MELLANOX_OFED_VERSION=4.9-2.2.4.0
+            MELLANOX_OFED_VERSION=4.9-4.1.7.0
             ;;
         *)
             # nvidia seems to make future DGX MOFED-less, and DGX OS starting
@@ -23,7 +30,7 @@ install_mellanox_ofed() {
             # performance characteristics for jammy and above.
             return 0
     esac
-    mlnx_url="${MLNX_REPO}/${MELLANOX_OFED_VERSION}/ubuntu${ubuntu_ver}/mellanox_mlnx_ofed.list"
+    mlnx_url="${MLNX_REPO}/${MELLANOX_OFED_VERSION}/ubuntu${UBUNTU_VER}/mellanox_mlnx_ofed.list"
 
     wget -O - https://www.mellanox.com/downloads/ofed/RPM-GPG-KEY-Mellanox | \
         apt-key add -