Set NVIDIA GPU to TCC or WDDM

smi-set-nv.ps1

$smi_location = ""
$smi_exe = "nvidia-smi.exe"
$global:gpu_id = ""
$tcc_id = 1
$wddm_id = 0

function get_gpu_uuid() {
    Write-Output "Getting GPU UUID..."
    $global:gpu_id = cd $smi_location | .\nvidia-smi.exe --query | findstr "UUID"
    $global:gpu_id = $global:gpu_id -replace ".* "
    $global:gpu_id = $global:gpu_id.Trim()
    Write-Output "Done, UUID is: $global:gpu_id"
}

function get_smi_dir() {
    Write-Output "Checking for nvidia-sli.exe existence in C:\..."
    Get-ChildItem -Path C:\ -Filter $smi_exe -Recurse -ErrorAction SilentlyContinue -Force | findstr -i Directory
    $smi_location = Read-Host "Please copy/paste a directory found above (if any): > "
    $smi_location = $smi_location.Trim() # Just incase of any whitespace during paste.
}

function set_tcc() {
    Write-Output "Setting $global:gpu_id to TCC mode."
    cd $smi_location | .\nvidia-smi.exe -g $global:gpu_id -dm $tcc_id
}

function set_wddm() {
    Write-Host "Setting $global:gpu_id to WDDM mode."
    cd $smi_location | .\nvidia-smi.exe -g $global:gpu_id -dm $wddm_id
}

function main() {
    $tcc_or_wddm = Read-Host "Would you like to set your GPU to TCC or WDDM? [tcc/wddm] >> "
    if ($tcc_or_wddm -eq "tcc") {
        get_smi_dir
        get_gpu_uuid
        set_tcc
    }
    if ($tcc_or_wddm -eq "wddm") {
        get_smi_dir
        get_gpu_uuid
        set_wddm
    }
}

main