From 1704f7c52113e3acdba605c8188c9e0a1a1ca460 Mon Sep 17 00:00:00 2001
From: "Ward Nakchbandi (Cosmic Fusion)"
 <83735213+CosmicFusion@users.noreply.github.com>
Date: Sun, 28 Jan 2024 22:01:33 +0300
Subject: [PATCH] Update main.sh

---
 main.sh | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.sh b/main.sh
index 8b87a1d..94f0a93 100755
--- a/main.sh
+++ b/main.sh
@@ -1,7 +1,9 @@
 #! /bin/bash
 DRIVER=545
 
-apt show nvidia-driver-$DRIVER 2>&1 | grep -v "does not have a stable" | grep Version: | head -n1 | cut -f2 -d":" | tr -d ' ' > ./linux-nvidia-modules/pika_nvidia.txt
+echo "$(apt show kernel-pika 2>&1 | grep -v "does not have a stable" | grep Depends: | head -n1 | cut -f2 -d":" | cut -f1 -d"," | cut -f3,4 -d"-" | tr -d ' ')" > ./linux-nvidia-modules/KERNEL
+
+apt show linux-modules-nvidia-$DRIVER-$(./linux-nvidia-modules/KERNEL) 2>&1 | grep -v "does not have a stable" | grep Version: | head -n1 | cut -f2 -d":" | tr -d ' ' > ./linux-nvidia-modules/pika_nvidia.txt
 
 rm -rfv /etc/apt/preferences.d/*
 echo 'Pin: release c=external' > /etc/apt/preferences.d/0-a