Hyper-V | Assign Graphics Card to Virtual Machine

# Define Variables
$VMName = "WR-CLI-VM-MPS"
$LogPath = "C:\Logs"

# Ensure the log directory exists
if (!(Test-Path -Path $LogPath)) {
    New-Item -ItemType Directory -Path $LogPath -Force | Out-Null
}

$OutputFile = "$LogPath\discrete-device-assignment-mount.txt"

# Function to log and output messages
function Log-Message {
    param ($Message)
    Write-Output $Message | Tee-Object -FilePath $OutputFile -Append
}

# Configure VM for Discrete Device Assignment
Set-VM -Name $VMName -AutomaticStopAction TurnOff
Set-VM -GuestControlledCacheTypes $True -VMName $VMName
Set-VM -LowMemoryMappedIoSpace 3GB -VMName $VMName
Set-VM -HighMemoryMappedIoSpace 33280MB -VMName $VMName

# Get all NVIDIA Display Devices
$GPUDevices = Get-PnpDevice -PresentOnly | Where-Object { 
    $_.Class -eq "Display" -and $_.Manufacturer -like "NVIDIA" 
}

if ($GPUDevices.Count -eq 0) {
    Log-Message "No NVIDIA GPU found for passthrough. Exiting."
    exit
}

# Select the first available NVIDIA GPU and retrieve its location path
$SelectedGPU = $GPUDevices[0]
$LocationPath = ($SelectedGPU | Get-PnpDeviceProperty -KeyName 'DEVPKEY_Device_LocationPaths').Data[0]

# Log selected GPU information
Log-Message "Selected GPU: $($SelectedGPU.FriendlyName)"
Log-Message "Location Path: $LocationPath"

# Disable GPU on Host
Disable-PnpDevice -InstanceId $SelectedGPU.InstanceId -Confirm:$False

# Dismount GPU from Host
Dismount-VMHostAssignableDevice -Force -LocationPath $LocationPath

# Assign GPU to VM
Add-VMAssignableDevice -LocationPath $LocationPath -VMName $VMName

# Final Output
Log-Message "GPU successfully assigned to $VMName."

# Define Variables
$VMName = "WR-CLI-VM-MPS"
$LogPath = "C:\Logs"

# Ensure log directory exists
if (!(Test-Path -Path $LogPath)) {
    New-Item -ItemType Directory -Path $LogPath -Force | Out-Null
}

$OutputFile = "$LogPath\device-reassignment-log.txt"

# Function to log messages
function Log-Message {
    param ($Message)
    Write-Output $Message | Tee-Object -FilePath $OutputFile -Append
}

# Enumerate all PnP Devices and find NVIDIA GPUs
$GPUDevices = Get-PnpDevice | Where-Object { 
    $_.Class -eq "Display" -and $_.Manufacturer -like "NVIDIA" 
}

if ($GPUDevices.Count -eq 0) {
    Log-Message "No NVIDIA GPU found for reattachment. Exiting."
    exit
}

# Select the first available NVIDIA GPU and retrieve its location path
$SelectedGPU = $GPUDevices[0]
$LocationPath = ($SelectedGPU | Get-PnpDeviceProperty -KeyName 'DEVPKEY_Device_LocationPaths').Data[0]

# Log selected GPU information
Log-Message "Selected GPU: $($SelectedGPU.FriendlyName)"
Log-Message "Location Path: $LocationPath"

# Remove the device from the VM
try {
    Remove-VMAssignableDevice -LocationPath $LocationPath -VMName $VMName -ErrorAction Stop
    Log-Message "Successfully removed GPU from VM: $VMName."
} catch {
    Log-Message "ERROR: Failed to remove GPU from VM: $_"
    exit
}

# Mount the device back to the host
try {
    Mount-VMHostAssignableDevice -LocationPath $LocationPath -ErrorAction Stop
    Log-Message "Successfully mounted GPU back to the host."
} catch {
    Log-Message "ERROR: Failed to mount GPU back to the host: $_"
    exit
}

# Enable the GPU on the host
try {
    Enable-PnpDevice -InstanceId $SelectedGPU.InstanceId -Confirm:$False -ErrorAction Stop
    Log-Message "Successfully enabled GPU on the host."
} catch {
    Log-Message "ERROR: Failed to enable GPU on the host: $_"
    exit
}

# Final Output
Log-Message "GPU reattached and enabled successfully."