Resetting gpu

reset_gpu.sh

#!/bin/bash

unbind_gpu() {
    echo "Unbinding NVIDIA driver..."
    GPU_PCI=$(lspci | grep -i nvidia | cut -d ' ' -f 1)
    for gpu in $GPU_PCI; do
        echo -n "0000:$gpu" > /sys/bus/pci/drivers/nvidia/unbind
    done
}

remove_nvidia_modules() {
    echo "Removing NVIDIA kernel modules..."
    modprobe -r nvidia_drm nvidia_uvm nvidia_modeset nvidia
}

load_nvidia_modules() {
    echo "Loading NVIDIA kernel modules..."
    modprobe nvidia
    modprobe nvidia_modeset
    modprobe nvidia_uvm
    modprobe nvidia_drm
}

bind_gpu() {
    echo "Binding NVIDIA driver..."
    GPU_PCI=$(lspci | grep -i nvidia | cut -d ' ' -f 1)
    for gpu in $GPU_PCI; do
        echo -n "0000:$gpu" > /sys/bus/pci/drivers/nvidia/bind
    done
}

unbind_gpu
remove_nvidia_modules
load_nvidia_modules
bind_gpu

echo "GPU reset complete."