#!/bin/bash
modprobe vfio-pci
for dev in "$@"; do
    vendor=$(cat /sys/bus/pci/devices/$dev/vendor)
    device=$(cat /sys/bus/pci/devices/$dev/device)
    echo $vendor $device > /sys/bus/pci/drivers/vfio-pci/new_id
    if [ -e /sys/bus/pci/devices/$dev/driver ]; then
	    rmmod nvidia_drm
	    rmmod nvidia_modeset
	    rmmod nvidia
	    echo $dev > /sys/bus/pci/devices/$dev/driver/unbind
	    echo "Unbound old driver"
    fi
    echo $vendor $device > /sys/bus/pci/drivers/vfio-pci/new_id
    echo "Bound VFIO driver"
done