mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
52 lines
1.6 KiB
PowerShell
52 lines
1.6 KiB
PowerShell
$defaultEnv = @()
|
|
foreach($line in Get-Content -Path $env:LOCALAPPDATA\trt_env_outlog.txt) {
|
|
$defaultEnv += $line
|
|
}
|
|
|
|
#The order of data is:
|
|
# CUDA Status: 1 present (preceded by version), 0 not present (preceded by message)
|
|
# Python: 0 not present, 1 present at 3.10
|
|
# Microsoft MPI: 0 not present, 1 present
|
|
# Microsoft MPI in EnvPath: 0 not present, 1 present
|
|
|
|
#0 = CUDA
|
|
#2 = CUDA Version
|
|
#3 = Python
|
|
#4 = MPI Presence
|
|
#5 = CUDNN
|
|
#6 = TRT
|
|
|
|
if ($defaultEnv[0].Equals("0")) {
|
|
Write-Output "Removing CUDA"
|
|
Uninstall-Package -Name $defaultEnv[1]
|
|
}
|
|
|
|
if ($defaultEnv[2].Equals("0")) {
|
|
Write-Output "Removing Python 3.10.11"
|
|
Uninstall-Package -Name "Python 3.10.11 (64-bit)"
|
|
}
|
|
|
|
if ($defaultEnv[3].Equals("0")) {
|
|
#MPI must be removed
|
|
Write-Output "Removing MPI"
|
|
uninstall-package -ProviderName msi -Name "Microsoft MPI SDK (10.1.12498.16)"
|
|
uninstall-package -ProviderName msi -Name "Microsoft MPI (10.1.12498.16)"
|
|
}
|
|
|
|
if ($defaultEnv[4].Equals("0")) {
|
|
#MPI must be removed from PATH
|
|
$path = [Environment]::GetEnvironmentVariable('path', 'Machine')
|
|
$path = ($path.Split(';') | Where-Object { $_ -ne 'C:\Program Files\Microsoft MPI\Bin' }) -join ';'
|
|
[System.Environment]::SetEnvironmentVariable("path", $path,'Machine')
|
|
}
|
|
|
|
if ($defaultEnv[5].Equals("0")) {
|
|
Write-Output "Removing CUDNN"
|
|
[Environment]::SetEnvironmentVariable('CUDNN', '', [EnvironmentVariableTarget]::Machine)
|
|
}
|
|
|
|
if ($defaultEnv[6].Equals("0")) {
|
|
Write-Output "Removing TRT"
|
|
[Environment]::SetEnvironmentVariable('TRT', '', [EnvironmentVariableTarget]::Machine)
|
|
}
|