Initial commit
This commit is contained in:
53
geniusia2/install_faiss.sh
Executable file
53
geniusia2/install_faiss.sh
Executable file
@@ -0,0 +1,53 @@
|
||||
#!/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
|
||||
Reference in New Issue
Block a user