Actualiser install.sh

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"