pre_upgrade() {
    pre_remove
}

post_upgrade() {
    echo ">>> Make sure to run sudo systemctl enable dkms.service, to have "
    echo ">>> this kernel module rebuild on each kernel update "
}

pre_remove() {
    dkms remove -m nvidia -v 313.18 --all
}