[ubuntu24.04] use --kernel-module-type open when specified

Signed-off-by: Tariq Ibrahim <[email protected]>

Author: tariq1890

ubuntu24.04/nvidia-driver

@@ -15,9 +15,7 @@ NVIDIA_UVM_MODULE_PARAMS=()
 NVIDIA_MODESET_MODULE_PARAMS=()
 NVIDIA_PEERMEM_MODULE_PARAMS=()
 TARGETARCH=${TARGETARCH:?"Missing TARGETARCH env"}
-
-OPEN_KERNEL_MODULES_ENABLED=${OPEN_KERNEL_MODULES_ENABLED:-false}
-[[ "${OPEN_KERNEL_MODULES_ENABLED}" == "true" ]] && KERNEL_TYPE=kernel-open || KERNEL_TYPE=kernel
+KERNEL_MODULE_TYPE=${KERNEL_MODULE_TYPE:-auto}
 
 export DEBIAN_FRONTEND=noninteractive
 
@@ -359,6 +357,10 @@ _install_driver() {
         install_args+=("--concurrency-level=${MAX_THREADS}")
     fi
 
+    if [[ "${KERNEL_MODULE_TYPE}" == "proprietary"  || "${KERNEL_MODULE_TYPE}" == "open" ]]; then
+        install_args+=("--kernel-module-type=${KERNEL_MODULE_TYPE}")
+    fi
+
     # Install the NVIDIA driver in one step
     sh NVIDIA-Linux-$DRIVER_ARCH-$DRIVER_VERSION.run --silent \
                     --ui=none \
@@ -374,7 +376,6 @@ _install_driver() {
                     --x-module-path=/tmp/null \
                     --x-library-path=/tmp/null \
                     --x-sysconfig-path=/tmp/null \
-                    -m="${KERNEL_TYPE}" \
                     ${install_args[@]+"${install_args[@]}"}
 }