54 lines
1.4 KiB
Bash
Executable File
54 lines
1.4 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Script d'installation rapide de FAISS pour RPA Vision V2
|
|
|
|
echo "=================================================="
|
|
echo "📦 Installation de FAISS"
|
|
echo "=================================================="
|
|
echo ""
|
|
|
|
# Vérifier que l'environnement virtuel existe
|
|
if [ ! -d "venv" ]; then
|
|
echo "❌ Environnement virtuel non trouvé"
|
|
echo " Exécutez d'abord: ./setup.sh"
|
|
exit 1
|
|
fi
|
|
|
|
# Activer l'environnement virtuel
|
|
source venv/bin/activate
|
|
|
|
# Vérifier si FAISS est déjà installé
|
|
if python -c "import faiss" 2>/dev/null; then
|
|
echo "✅ FAISS est déjà installé"
|
|
python -c "import faiss; print(f' Version: {faiss.__version__}')"
|
|
exit 0
|
|
fi
|
|
|
|
echo "📥 Installation de FAISS..."
|
|
echo ""
|
|
|
|
# Détecter si GPU est disponible
|
|
if python -c "import torch; print(torch.cuda.is_available())" 2>/dev/null | grep -q "True"; then
|
|
echo "🎮 GPU détecté - Installation de faiss-gpu"
|
|
pip install faiss-gpu
|
|
else
|
|
echo "💻 Pas de GPU - Installation de faiss-cpu"
|
|
pip install faiss-cpu
|
|
fi
|
|
|
|
# Vérifier l'installation
|
|
if python -c "import faiss" 2>/dev/null; then
|
|
echo ""
|
|
echo "✅ FAISS installé avec succès!"
|
|
python -c "import faiss; print(f' Version: {faiss.__version__}')"
|
|
|
|
echo ""
|
|
echo "📊 Prochaine étape: Reconstruire l'index"
|
|
echo " cd .."
|
|
echo " python3 rebuild_faiss_simple.py"
|
|
else
|
|
echo ""
|
|
echo "❌ Erreur lors de l'installation de FAISS"
|
|
exit 1
|
|
fi
|