#!/bin/bash
set -e
echo "Setting up NVIDIA GPU for Gazebo..."
if ! command -v nvidia-smi &> /dev/null; then
    echo "NVIDIA driver not found. Please install NVIDIA drivers first."
    exit 1
fi
nvidia-smi
export __NV_PRIME_RENDER_OFFLOAD=1
export __GLX_VENDOR_LIBRARY_NAME=nvidia
export LIBGL_ALWAYS_SOFTWARE=0
cat >> ~/.bashrc << 'EOF'
export __NV_PRIME_RENDER_OFFLOAD=1
export __GLX_VENDOR_LIBRARY_NAME=nvidia
export LIBGL_ALWAYS_SOFTWARE=0
EOF
echo ""
echo "NVIDIA setup complete!"
echo "Please restart your terminal or run: source ~/.bashrc"
echo ""
echo "Test with: glxinfo | grep 'OpenGL renderer'"