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

setup.ps1: fix GPU check to run nvidia-smi instead of just checking command existence

Browse source
This commit is contained in:
Roland Tannous 2026-03-14 19:51:08 +00:00
commit 87f60cd8f3
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
studio/setup.ps1
View file

@ -217,7 +217,11 @@ Write-Host "+==============================================+" -ForegroundColor G
# ============================================
# 1a. GPU requirement check
# ============================================
$HasNvidiaSmi = $null -ne (Get-Command nvidia-smi -ErrorAction SilentlyContinue)
$HasNvidiaSmi = $false
try {
nvidia-smi 2>&1 | Out-Null
if ($LASTEXITCODE -eq 0) { $HasNvidiaSmi = $true }
} catch {}
if (-not $HasNvidiaSmi) {
Write-Host ""
Write-Host "[ERROR] Unsloth Studio requires an NVIDIA GPU." -ForegroundColor Red