add nvidia sources for nvidia-docker

scripts/install.sh

@@ -138,6 +138,17 @@ no_suspend() {
     $SUDO systemctl restart systemd-logind.service
 }
 
+install_nvidia_docker() {
+    # get latest driver from https://www.nvidia.com/object/unix.html
+    # https://github.com/NVIDIA/nvidia-docker/wiki/Installation-(version-2.0)#prerequisites
+    curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
+            sudo tee /etc/apt/sources.list.d/nvidia-docker.list
+            sudo apt-get update
+
+            # Install nvidia-docker2 and reload the Docker daemon configuration
+            sudo apt-get install -y nvidia-docker2
+            sudo pkill -SIGHUP dockerd
+}
 install_i3() {
     echo "update and installing i3wm and some tools..."