update

check-gpu.sh

@@ -0,0 +1,54 @@
+#!/bin/bash
+
+# Script to check GPU availability for Ollama
+
+echo "GPU Availability Check"
+echo "======================"
+echo ""
+
+# Check for NVIDIA GPU
+if command -v nvidia-smi &> /dev/null; then
+    echo "✓ NVIDIA GPU detected"
+    echo ""
+    echo "GPU Information:"
+    nvidia-smi --query-gpu=index,name,driver_version,memory.total,memory.free --format=csv,noheader | \
+        awk -F', ' '{printf "  GPU %s: %s\n  Driver: %s\n  Memory: %s total, %s free\n\n", $1, $2, $3, $4, $5}'
+    
+    # Check CUDA version
+    if command -v nvcc &> /dev/null; then
+        echo "CUDA Version:"
+        nvcc --version | grep "release" | awk '{print "  " $0}'
+        echo ""
+    fi
+    
+    # Check Docker GPU support
+    echo "Checking Docker GPU support..."
+    if docker run --rm --gpus all nvidia/cuda:12.0.0-base-ubuntu22.04 nvidia-smi &> /dev/null; then
+        echo "✓ Docker can access GPU"
+        echo ""
+        echo "Recommendation: Use GPU-accelerated startup"
+        echo "  ./start-with-gpu.sh"
+    else
+        echo "✗ Docker cannot access GPU"
+        echo ""
+        echo "Install NVIDIA Container Toolkit:"
+        echo "  https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html"
+        echo ""
+        echo "After installation, restart Docker:"
+        echo "  sudo systemctl restart docker"
+    fi
+else
+    echo "ℹ No NVIDIA GPU detected"
+    echo ""
+    echo "Running Ollama on CPU is supported but slower."
+    echo ""
+    echo "Performance comparison:"
+    echo "  CPU: ~1-2s per translation, ~8s per summary"
+    echo "  GPU: ~0.3s per translation, ~2s per summary"
+    echo ""
+    echo "Recommendation: Use standard startup"
+    echo "  docker-compose up -d"
+fi
+
+echo ""
+echo "For more information, see: docs/OLLAMA_SETUP.md"