/
Chaizee
/
ZenithCode_Incident-LLM-analytics
Обзор
Документация
Войти
/
Chaizee
/
ZenithCode_Incident-LLM-analytics
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
scripts/install_gpu.ps1
29 строк
1 KB
Chaizee
fix: fix speed
12 июн 2026, 16:28
12 июн 2026, 16:28
b2745cc
Код
Авторство
О чём код?
#Requires -Version 5.1 $ErrorActionPreference = "Stop" Set-Location $PSScriptRoot\.. Write-Host "=== GPU setup: PyTorch CUDA 12.4 + llama-cpp-python CUDA ===" -ForegroundColor Cyan if (-not (Test-Path ".venv\Scripts\python.exe")) { Write-Host "Creating venv..." python -m venv .venv } $pip = ".\.venv\Scripts\pip.exe" & $pip install --upgrade pip Write-Host "Installing base requirements..." & $pip install -r requirements.txt Write-Host "Installing CUDA PyTorch..." & $pip uninstall -y torch torchvision 2>$null & $pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124 Write-Host "Installing llama-cpp-python with CUDA..." & $pip uninstall -y llama-cpp-python 2>$null & $pip install llama-cpp-python --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu124 Write-Host "Verifying CUDA..." & .\.venv\Scripts\python.exe -c "import torch; print('cuda:', torch.cuda.is_available()); print(torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'NO GPU')" Write-Host "Done. Run: .\.venv\Scripts\streamlit.exe run app.py" -ForegroundColor Green