NVIDIA Driver Stopped Working

I have on laptop with 3070 TI
So Installed the CUDA 11.8. This installed me this nvidia-driver: nvidia/520.61.05

When I run


Got this:

NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.

Ran this:

dkms status

Got this:

nvidia/520.61.05, 5.19.0-32-generic, x86_64: installed

Tried this:

sudo update-initramfs -u

Got this:

update-initramfs: Generating /boot/initrd.img-6.2.0-26-generic
I: The initramfs will attempt to resume from /dev/nvme0n1p9
I: (UUID=32d51fe8-c309-4296-97a0-65e0aedf6dba)
I: Set the RESUME variable to override this.

How do I solve the issue? And also I do I keep the CUDA still working after the driver update?