feat: deteccion de GPU NVIDIA y aceleracion automatica

- Detecta nvidia-smi (driver Windows 535+ hace passthrough a WSL2)
- Con GPU: ollama arranca con --gpus all (docker run) / override
  compose docker-compose.gpu.yml (gpus: all) -> 30-50 tok/s
- Sin GPU: como hasta ahora (CPU ~10-20 tok/s) + aviso honesto
This commit is contained in:
hermes committed 2026-09-12 21:33:31 +02:00
1 parent 6f052cf738
commit 25970017fa
4 files changed
+72 -4

No files matched your search

+16 -1
View File
@@ -83,6 +83,19 @@ Ok "RAM: $TotalRamGb GB"
# --- Tumbar deploy previo (antes de detectar puertos) -----------------------
cmd /c docker rm -f d0a1-hermes-ollama '>' NUL '2>&1'
# --- GPU NVIDIA? (si hay, Ollama la usa y va 5-10x mas rapido) ---------------
$HasGpu = $false
if (Get-Command nvidia-smi -ErrorAction SilentlyContinue) {
$GpuInfo = & nvidia-smi --query-gpu=name,memory.total --format=csv,noheader 2>$null
if ($LASTEXITCODE -eq 0 -and $GpuInfo) {
$HasGpu = $true
$GpuName = ($GpuInfo -split ',')[0].Trim()
Info "GPU NVIDIA detectada: $GpuName (aceleracion activada)"
}
}
if (-not $HasGpu) {
Warn 'Sin GPU NVIDIA: modo CPU (~10-20 tokens/s). Con una GPU NVIDIA va 5-10x mas rapido.'
}
# --- Auto-detect Ollama port ------------------------------------------------
$OllamaPort = Find-FreePort 11434
@@ -101,7 +114,9 @@ Write-Host ''
Warn '[1/3] Arrancando Ollama (IA local)...'
$running = docker ps --format '{{.Names}}' | Select-String -Quiet 'd0a1-hermes-ollama'
if (-not $running) {
docker run -d --name d0a1-hermes-ollama --memory 6g `
$GpuArgs = @()
if ($HasGpu) { $GpuArgs = @('--gpus', 'all') }
docker run -d --name d0a1-hermes-ollama --memory 6g @GpuArgs `
-p "${OllamaPort}:11434" `
-v d0a1_hermes_ollama:/root/.ollama `
-e OLLAMA_HOST=0.0.0.0 `