From 87f60cd8f3f566f10cc3e0e3ba403eab76d2ecf6 Mon Sep 17 00:00:00 2001
From: Roland Tannous <rolandtannous@gravityq.ai>
Date: Sat, 14 Mar 2026 19:51:08 +0000
Subject: [PATCH] setup.ps1: fix GPU check to run nvidia-smi instead of just
 checking command existence

---
 studio/setup.ps1 | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/studio/setup.ps1 b/studio/setup.ps1
index 352319ce5e..63ce76151e 100644
--- a/studio/setup.ps1
+++ b/studio/setup.ps1
@@ -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