From 10731461dcc58b4ddc56f0047c782bd65b9f13d8 Mon Sep 17 00:00:00 2001
From: Francois B <shakasan@sirenacorp.be>
Date: Wed, 2 May 2018 01:25:14 +0200
Subject: [PATCH] nvidia396 added

---
 install_functions.sh | 14 ++++++++++++++
 menus_functions.sh   |  4 ++++
 repo_functions.sh    |  4 ++++
 yggdrasil.sh         | 13 +------------
 4 files changed, 23 insertions(+), 12 deletions(-)

diff --git a/install_functions.sh b/install_functions.sh
index 05bcfa0..d2cfb3c 100644
--- a/install_functions.sh
+++ b/install_functions.sh
@@ -529,6 +529,20 @@ function installNvidia390Menu () {
   installAppsFromListMenu nvidia-390
 }
 
+#
+# install Nvidia Drivers (headless)
+#
+function installNvidia396 () {
+  installAppsFromList nvidia-396
+}
+
+#
+# install Nvidia Drivers (Menu)
+#
+function installNvidia396Menu () {
+  installAppsFromListMenu nvidia-396
+}
+
 #
 # install TLP App (headless)
 #
diff --git a/menus_functions.sh b/menus_functions.sh
index 574203c..0528866 100644
--- a/menus_functions.sh
+++ b/menus_functions.sh
@@ -413,6 +413,7 @@ function showHardwareMenu () {
     "nvidia384" "Install Nvidia 384 graphic drivers" \
     "nvidia387" "Install Nvidia 387 graphic drivers" \
     "nvidia390" "Install Nvidia 390 graphic drivers" \
+    "nvidia396" "Install Nvidia 396 graphic drivers" \
     "tlp" "Install/Enable TLP for better power management " \
     "keyid" "Add udev rules for Key-ID FIDO U2F usb key" \
     "WI6320" "Fix Intel Wireless 6320 card config problem" \
@@ -452,6 +453,9 @@ function showHardwareMenu () {
       "nvidia390")
         installNvidia390Menu
         ;;
+      "nvidia396")
+        installNvidia396Menu
+        ;;
       "tlp")
         installTLPMenu
         ;;
diff --git a/repo_functions.sh b/repo_functions.sh
index e25f279..3f0bdae 100644
--- a/repo_functions.sh
+++ b/repo_functions.sh
@@ -343,6 +343,10 @@ nvidia-390;apt;nvidia-390;nvidia-390
 nvidia-settings;apt;nvidia-390;nvidia-settings
 nvidia-opencl-icd-390;apt;nvidia-390;nvidia-opencl-icd-390
 libcuda1-390;apt;nvidia-390;libcuda1-390
+nvidia-396;apt;nvidia-396;nvidia-396
+nvidia-settings;apt;nvidia-396;nvidia-settings
+nvidia-opencl-icd-396;apt;nvidia-396;nvidia-opencl-icd-396
+libcuda1-396;apt;nvidia-396;libcuda1-396
 tlp;apt;tlp;tlp
 notepadqq;apt;dev;notepadqq
 agave;apt;dev;agave
diff --git a/yggdrasil.sh b/yggdrasil.sh
index 1bbcbe5..ba57a57 100755
--- a/yggdrasil.sh
+++ b/yggdrasil.sh
@@ -52,17 +52,6 @@ echo "--[ Yggdrasil log ]--[ "$cDate" ]--[ "$cTime" ]-----------------------" >>
 
 while getopts ":h,v,f,c,u,a,d,q,s,t,n,w,p,k,T" option; do
   case "$option" in
-    T) #test purpose only
-      yggInit
-      #installIdea
-      #installNetTools
-      #installOffice
-      #addPPA "tmsu/ppa"
-      updateSystem
-      addRequiredPPA
-      installTeamViewer13
-      exit
-      ;;
     a) # install all apps
        # except
        # apps : Nitrogen, Beta apps, Nightly, Unboud
@@ -208,7 +197,7 @@ while getopts ":h,v,f,c,u,a,d,q,s,t,n,w,p,k,T" option; do
     n) # latest nvidia driver (see menus if need older version)
       yggInit
       updateSystem
-      installNvidia390
+      installNvidia396
       exit
       ;;
     k)