#!/bin/bash
set -x

# # Re-Bind GPU to Nvidia Driver
virsh nodedev-reattach pci_0000_01_00_1
virsh nodedev-reattach pci_0000_01_00_0


# Reload vfio
modprobe -r vfio_pci
modprobe -r vfio_iommu_type1
modprobe -r vfio


# # Rebind VT consoles
echo 1 > /sys/class/vtconsole/vtcon0/bind
# # Some machines might have more than 1 virtual console. Add a line for each corresponding VTConsole
echo 1 > /sys/class/vtconsole/vtcon1/bind

# nvidia-xconfig --query-gpu-info > /dev/null 2>&1
echo "efi-framebuffer.0" > /sys/bus/platform/drivers/efi-framebuffer/bind

# # Reload nvidia modules
modprobe nvidia
modprobe nvidia_modeset
modprobe nvidia_uvm
modprobe nvidia_drm

# # Restart Display Manager
systemctl start nvidia-persistenced.service
systemctl start display-manager.service