add open nvidia driver support for ubuntu

Signed-off-by: shiva kumar <[email protected]>

Author: shivakunv

ubuntu22.04/precompiled/Dockerfile

@@ -65,7 +65,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-$
 # avoid cleaning the cache after this to retain these packages during runtime.
 RUN apt-get update && apt-get install --download-only --no-install-recommends -y linux-objects-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION} \
     linux-signatures-nvidia-${KERNEL_VERSION} \
-    linux-modules-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION}
+    linux-modules-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION} \
+    # add support for nvidia open source driver packages during runtime
+    linux-objects-nvidia-${DRIVER_BRANCH}-server-open-${KERNEL_VERSION} \
+    linux-modules-nvidia-${DRIVER_BRANCH}-server-open-${KERNEL_VERSION}
 
 COPY nvidia-driver /usr/local/bin
 

ubuntu22.04/precompiled/nvidia-driver

@@ -4,6 +4,7 @@
 set -eu
 
 KERNEL_VERSION=$(uname -r)
+OPEN_KERNEL_MODULES_ENABLED="${OPEN_KERNEL_MODULES_ENABLED:-false}"
 RUN_DIR=/run/nvidia
 PID_FILE=${RUN_DIR}/${0##*/}.pid
 DRIVER_BRANCH=${DRIVER_BRANCH:?"Missing driver version"}
@@ -210,10 +211,25 @@ _unload_driver() {
 
 # Link and install the kernel modules from a precompiled packages
 _install_driver() {
-    echo "Installing NVIDIA driver kernel modules..."
-    apt-get install --no-install-recommends -y linux-objects-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION} \
-    linux-signatures-nvidia-${KERNEL_VERSION} \
-	linux-modules-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION}
+    if [ "$OPEN_KERNEL_MODULES_ENABLED" = true ]; then
+        echo "Installing Open NVIDIA driver kernel modules..."
+        apt-get install -y  linux-signatures-nvidia-${KERNEL_VERSION}
+        # installing closed linux-modules-${KERNEL_VERSION} , as linux-objects-nvidia-${DRIVER_BRANCH}-server-open-${KERNEL_VERSION} 
+        # has dependency on it and it is auto installing it and has conflict with linux-modules-nvidia-${DRIVER_BRANCH}-server-open-${KERNEL_VERSION}
+        # https://forums.developer.nvidia.com/t/ubuntu-and-nvidia-provided-packages-conflict-breaking-installation/259150
+        apt-get install --download-only --no-install-recommends -y linux-objects-nvidia-${DRIVER_BRANCH}-server-open-${KERNEL_VERSION} && \
+            dpkg -i --force-overwrite $(find /var/cache/apt/archives/ -name "linux-objects-nvidia-${DRIVER_BRANCH}-server-open-${KERNEL_VERSION}_*.deb")
+        apt-get install --download-only --no-install-recommends -y linux-modules-nvidia-${DRIVER_BRANCH}-server-open-${KERNEL_VERSION} && \
+            dpkg -i --force-overwrite $(find /var/cache/apt/archives/ -name "linux-modules-nvidia-${DRIVER_BRANCH}-server-open-${KERNEL_VERSION}_*.deb")
+        # Ensure modules are correctly loaded
+        depmod -a ${KERNEL_VERSION}
+    else
+        echo "Installing Closed NVIDIA driver kernel modules..."
+        apt-get install --no-install-recommends -y \
+            linux-objects-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION} \
+            linux-signatures-nvidia-${KERNEL_VERSION} \
+            linux-modules-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION}
+    fi
 }
 
 # Mount the driver rootfs into the run directory with the exception of sysfs.