[feature] add auto install driver for ubuntu

2021-01-04 16:25:17 +08:00 · 2021-01-04 16:25:17 +08:00 · f10044bbaf
commit f10044bbaf
parent 345ea0f46d
1 changed files with 2 additions and 0 deletions
--- 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