ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Always persist compatible CUDA_PATH to User registry (overwrite stale values)

Browse source
This commit is contained in:
Roland Tannous 2026-02-28 07:51:31 +00:00
commit 7b4d074857
1 changed files with 4 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
setup.ps1
View file

@ -381,13 +381,10 @@ $CudaToolkitRoot = Split-Path (Split-Path $NvccPath -Parent) -Parent
# CudaToolkitDir: the MSBuild property that CUDA .targets checks directly
# Trailing backslash required -- the .targets file appends subpaths to it
[Environment]::SetEnvironmentVariable('CudaToolkitDir', "$CudaToolkitRoot\", 'Process')
# Persist CUDA_PATH to User registry if not already set
$existingSys = [Environment]::GetEnvironmentVariable('CUDA_PATH', 'Machine')
$existingUsr = [Environment]::GetEnvironmentVariable('CUDA_PATH', 'User')
if (-not $existingSys -and -not $existingUsr) {
[Environment]::SetEnvironmentVariable('CUDA_PATH', $CudaToolkitRoot, 'User')
Write-Host " Persisted CUDA_PATH to user environment" -ForegroundColor Gray
}
# Always persist CUDA_PATH to User registry so the compatible toolkit is used
# in future sessions (overwrites any existing value pointing to a newer, incompatible version)
[Environment]::SetEnvironmentVariable('CUDA_PATH', $CudaToolkitRoot, 'User')
Write-Host " Persisted CUDA_PATH=$CudaToolkitRoot to user environment" -ForegroundColor Gray
# Ensure nvcc's bin dir is on PATH for this process
$nvccBinDir = Split-Path $NvccPath -Parent
if ($env:PATH -notlike "*$nvccBinDir*") {