#!/bin/sh
[ "$(id -u)" = "0" ] || exit 1
[ -x /usr/bin/nvidia-installer ] && nvidia-installer --uninstall --skip-module-unload --silent
DISTRO=$(lsb_release -is|tr [:upper:] [:lower:])$(lsb_release -rs)
ARCH=$(uname -m)
TMP=$(mktemp -d /tmp/cuda.XXXXX)
wget -NP $TMP https://developer.download.nvidia.com/compute/cuda/repos/$DISTRO/$ARCH/cuda-keyring_1.1-1_all.deb
dpkg -i $TMP/cuda-keyring_1.1-1_all.deb
rm -r $TMP
apt update
apt install --yes cuda-toolkit
apt install --yes --install-recommends nvidia-driver nvidia-settings nvidia-kernel-open-dkms
apt install --yes nvidia-driver-cuda

# this disables the splash and forces repair on every boot - introduces a delay that could be useful
# not always needed - best disable splash in kde as well if you enable it
if  [ "$1" = "--nosplash" ]; then
	sed -i 's/^\(GRUB_CMDLINE_LINUX_DEFAULT=\).*/\1"quiet fsck.mode=force fsck.repair=yes"/' /etc/default/grub
	update-grub
fi