From ff3327bae13ae1e9175d3eb931fc49a2375d59ab Mon Sep 17 00:00:00 2001
From: wilou <wilfried@sympatech.fr>
Date: Fri, 11 Oct 2024 12:34:25 +0000
Subject: [PATCH] Actualiser install.sh

---
 install.sh | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/install.sh b/install.sh
index ead327e..eeb9f11 100644
--- a/install.sh
+++ b/install.sh
@@ -274,9 +274,6 @@ unzip $builddir/themes.zip -d /etc/skel/.themes/
 rm -r $builddir
 rm -r /tmp/wibian/
 
-# Nettoyage des paquets inutiles du système
-apt autoremove 
-
 # Activation de la gestion des interfaces dans NetworkManager
 sed -i 's/false/true/g' /etc/NetworkManager/NetworkManager.conf
 sed -i '9,$d' /etc/network/interfaces
@@ -289,6 +286,21 @@ sed -i 's/^PERCENT=50/PERCENT=25/' /etc/default/zramswap
 # Ajout de l'utilisateur principal dans le groupe lpadmin pour effectuer des tâches administratives liées à CUPS
 usermod -aG lpadmin $username
 
+# Installation des drivers Nvidia (reboot nécessaire)
+gpu=$(lspci | grep -i 'nvidia')
+
+if [[ -n $gpu ]]; then
+  printf 'Une carte Nvidia a été détectée: %s\n' "$gpu"
+  apt install -y linux-headers-amd64
+  apt install -y firmware-misc-nonfree nvidia-driver nvidia-smi
+  apt install -y nvidia-cuda-dev nvidia-cuda-toolkit
+else
+  printf 'Aucune carte Nvidia détectée\n'
+fi
+
+# Nettoyage des paquets inutiles du système
+apt autoremove 
+
 # Message de fin
 echo "Bisous" | figlet -f big
 echo -ne "\n"