post_upgrade() {
  echo "If you run into trouble with CUDA not being available, run nvidia-modprobe first."
}

post_install() {
  post_upgrade
}