From f10044bbafe00e871c5a7e66cb329be386beaab7 Mon Sep 17 00:00:00 2001
From: Klesh Wong <klesh@kleshwong.com>
Date: Mon, 4 Jan 2021 16:25:17 +0800
Subject: [PATCH] [feature] add auto install driver for ubuntu

---
 devel/deeplearning.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/devel/deeplearning.sh b/devel/deeplearning.sh
index d96e1c9..e7b64b4 100755
--- a/devel/deeplearning.sh
+++ b/devel/deeplearning.sh
@@ -9,6 +9,8 @@ DIR=$(dirname "$(readlink -f "$0")")
 # install deep learning tools
 case "$PM" in
     apt)
+        # auto install nvidia driver
+        sudo ubuntu-drivers autoinstall
         # cuda
         . /etc/lsb-release
         if [ "$DISTRIB_RELEASE" = "18.04" ] && [ "$(uname -m)" = 'x86_64' ]; then