From e4a1dbe9f40ab5e9fe0142651a60c6bff3bddf7e Mon Sep 17 00:00:00 2001 From: Krzysztof kuhy Rudnicki Date: Fri, 27 Jun 2025 13:35:28 +0200 Subject: [PATCH] feat: automatically add for sudo when needed --- scripts/nvidia_troubleshoot.sh | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/scripts/nvidia_troubleshoot.sh b/scripts/nvidia_troubleshoot.sh index 2066ede..481360d 100755 --- a/scripts/nvidia_troubleshoot.sh +++ b/scripts/nvidia_troubleshoot.sh @@ -5,16 +5,23 @@ set -e # Exit on any error +# Function to check and request sudo privileges +check_sudo() { + if [[ $EUID -ne 0 ]]; then + echo "This script requires sudo privileges to modify system files." + echo "Requesting sudo access..." + exec sudo "$0" "$@" + fi +} + +# Check for sudo privileges first +check_sudo "$@" + echo "NVIDIA Comprehensive Troubleshooter & GSP Disabler" echo "==================================================" echo "Current Date: $(date)" echo "User: $USER" - -# Check if running as root -if [[ $EUID -ne 0 ]]; then - echo "Error: This script must be run as root (use sudo)" - exit 1 -fi +echo "Original user: ${SUDO_USER:-$USER}" # Check if nvidia module is loaded if ! lsmod | grep -q nvidia; then