From c3941d1f0b7bad3ba9ad1b3c0f4bf4ff2eac0328 Mon Sep 17 00:00:00 2001
From: alansari <terminal.v@gmail.com>
Date: Sun, 28 Jan 2024 08:12:21 +0300
Subject: [PATCH] Quick patch for v550+ drivers

---
 overlay/etc/cont-init.d/60-configure_gpu_driver.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/overlay/etc/cont-init.d/60-configure_gpu_driver.sh b/overlay/etc/cont-init.d/60-configure_gpu_driver.sh
index 8c4be98..8460abc 100644
--- a/overlay/etc/cont-init.d/60-configure_gpu_driver.sh
+++ b/overlay/etc/cont-init.d/60-configure_gpu_driver.sh
@@ -51,6 +51,10 @@ function install_nvidia_driver {
         # Download the driver (if it does not yet exist locally)
         download_driver
 
+        if (($(echo $nvidia_host_driver_version | cut -d '.' -f 1) >= 550)); then
+            sed -i 's/--no-multigpu//g' /etc/cont-init.d/70-configure_xorg.sh
+        fi
+
         if (($(echo $nvidia_host_driver_version | cut -d '.' -f 1) > 500)); then
             print_step_header "Installing NVIDIA driver v${nvidia_host_driver_version:?} to match what is running on the host"
             chmod +x "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run"