sudo apt-get purge nvidia-*
sudo apt-get update
sudo apt-get autoremove

# then stop and start the VM again

curl https://raw.githubusercontent.com/GoogleCloudPlatform/compute-gpu-installation/main/linux/install_gpu_driver.py --output install_gpu_driver.py
sudo python3 install_gpu_driver.py