#!/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

# Install NVIDIA driver (auto-select)
sudo ubuntu-drivers install --gpgpu

echo "Rebooting so nouveau is disabled and NVIDIA driver loads..."
sudo reboot