diff --git a/01-nvidia_driver_install_v2.sh b/01-nvidia_driver_install_v2.sh
index 4259599..f646d8b 100644
--- a/01-nvidia_driver_install_v2.sh
+++ b/01-nvidia_driver_install_v2.sh
@@ -1,8 +1,15 @@
 #!/bin/bash
 set -euo pipefail
+
 sudo tee /etc/modprobe.d/blacklist-nouveau.conf >/dev/null <<'EOF'
 blacklist nouveau
 options nouveau modeset=0
 EOF
+
 sudo update-initramfs -u
-sudo reboot
+
+# Install NVIDIA driver (auto-select)
+sudo ubuntu-drivers install --gpgpu
+
+echo "Rebooting so nouveau is disabled and NVIDIA driver loads..."
+sudo reboot
\ No newline at end of file