51 lines
1.4 KiB
Bash
51 lines
1.4 KiB
Bash
|
#!/bin/bash
|
||
|
|
# ARC Engine v3.0 Quick Start
|
||
|
|
# ============================
|
||
|
|
|
||
|
|
echo "🤖 ARC Engine v3.0 - Quick Start"
|
||
|
|
echo "================================="
|
||
|
|
|
||
|
|
# Check Python version
|
||
|
|
PYTHON_VERSION=$(python3 --version 2>&1 | cut -d' ' -f2 | cut -d'.' -f1,2)
|
||
|
|
echo "Python version: $PYTHON_VERSION"
|
||
|
|
|
||
|
|
# Check if requirements are installed
|
||
|
|
echo ""
|
||
|
|
echo "Checking dependencies..."
|
||
|
|
|
||
|
|
if ! python3 -c "import torch" 2>/dev/null; then
|
||
|
|
echo "⚠️ PyTorch not found. Installing core dependencies..."
|
||
|
|
pip install torch transformers accelerate peft bitsandbytes datasets trl safetensors tqdm matplotlib requests
|
||
|
|
fi
|
||
|
|
|
||
|
|
if python3 -c "import torch" 2>/dev/null; then
|
||
|
|
echo "✓ PyTorch installed"
|
||
|
|
else
|
||
|
|
echo "❌ PyTorch installation failed"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
if python3 -c "import transformers" 2>/dev/null; then
|
||
|
|
echo "✓ Transformers installed"
|
||
|
|
else
|
||
|
|
echo "❌ Transformers not found"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Check GPU
|
||
|
|
echo ""
|
||
|
|
echo "Checking GPU..."
|
||
|
|
python3 -c "import torch; print(f'CUDA available: {torch.cuda.is_available()}'); print(f'Device: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else \"CPU\"}') if torch.cuda.is_available() else None"
|
||
|
|
|
||
|
|
# Run the engine
|
||
|
|
echo ""
|
||
|
|
echo "================================="
|
||
|
|
echo "Starting ARC Engine v3.0..."
|
||
|
|
echo "================================="
|
||
|
|
echo ""
|
||
|
|
echo "First run will download the model (~16GB)"
|
||
|
|
echo "Type 'help' for commands, 'help <topic>' for specific help"
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
python3 arc_engine_v30_fixed.py
|