Update 01-nvidia_driver_install_v2.sh

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