Linux kernel post-installation script for re-installing the NVIDIA (CUDA) driver.

update-nvidia-driver.sh

#!/bin/bash

set -eu

KERNEL_DRIVER="/lib/modules/$1/kernel/drivers/video/nvidia.ko"
INSTALLER="/usr/src/cuda-toolkit"

# Build new driver if it doesn't exist
if [ -e $KERNEL_DRIVER ] ; then
  echo "NVIDIA driver already installed for this kernel" >&2
  exit 0
fi

echo "Re-installing NVIDIA driver for kernel $1..." >&2
sh $INSTALLER --silent --driver --no-opengl-libs 2>1 > /dev/null

if [ -e $KERNEL_DRIVER ] ; then
  echo "Driver succesfully installed for kernel $1" >&2
else
  echo "Driver installation failed! See /var/log/nvidia-installer.log" >&2
fi