/
Chaizee
/
ZenithCode_Incident-LLM-analytics
Обзор
Документация
Войти
/
Chaizee
/
ZenithCode_Incident-LLM-analytics
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
scripts/install_gpu.sh
22 строки
817 B
Chaizee
fix: fix speed
12 июн 2026, 16:28
12 июн 2026, 16:28
b2745cc
Код
Авторство
О чём код?
#!/usr/bin/env bash set -euo pipefail cd "$(dirname "$0")/.." echo "=== GPU setup: PyTorch CUDA 12.4 + llama-cpp-python CUDA ===" if [[ ! -x .venv/bin/python ]]; then python3 -m venv .venv fi .venv/bin/pip install --upgrade pip .venv/bin/pip install -r requirements.txt .venv/bin/pip uninstall -y torch torchvision 2>/dev/null || true .venv/bin/pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124 .venv/bin/pip uninstall -y llama-cpp-python 2>/dev/null || true .venv/bin/pip install llama-cpp-python --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu124 .venv/bin/python -c "import torch; print('cuda:', torch.cuda.is_available()); print(torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'NO GPU')" echo "Done. Run: .venv/bin/streamlit run app.py"