llm_lab_perso/setup_env.sh
2025-03-27 18:40:52 +01:00

85 lines
2.5 KiB
Bash

#!/bin/bash
# Script de configuration de l'environnement virtuel LLM Lab
# Ce script crée un nouvel environnement virtuel et installe les dépendances requises
echo "=== Configuration de l'environnement LLM Lab ==="
# Vérification si Python 3 est installé
if ! command -v python3 &> /dev/null; then
echo "Erreur: Python 3 n'est pas installé. Veuillez l'installer avant de continuer."
exit 1
fi
# Vérification si pip est installé
if ! command -v pip3 &> /dev/null; then
echo "Erreur: pip3 n'est pas installé. Veuillez l'installer avant de continuer."
exit 1
fi
# Vérification si venv est disponible
python3 -c "import venv" &> /dev/null
if [ $? -ne 0 ]; then
echo "Le module venv n'est pas disponible. Installation en cours..."
sudo apt-get update
sudo apt-get install -y python3-venv
fi
# Vérification de l'installation de tkinter
python3 -c "import tkinter" &> /dev/null
if [ $? -ne 0 ]; then
echo "Tkinter n'est pas installé. Installation en cours..."
sudo apt-get update
sudo apt-get install -y python3-tk
fi
# Suppression de l'ancien environnement s'il existe
if [ -d "llmlab" ]; then
echo "Suppression de l'ancien environnement virtuel..."
rm -rf llmlab
fi
# Création du nouvel environnement virtuel
echo "Création d'un nouvel environnement virtuel..."
python3 -m venv llmlab
# Activation de l'environnement virtuel
echo "Activation de l'environnement virtuel..."
source llmlab/bin/activate
# Mise à jour de pip
echo "Mise à jour de pip..."
pip install --upgrade pip
# Installation des dépendances
echo "Installation des dépendances requises..."
pip install -r requirements.txt
# Installation optionnelle de dépendances pour NVIDIA GPU
if command -v nvidia-smi &> /dev/null; then
echo "NVIDIA GPU détecté, installation des dépendances NVIDIA..."
pip install nvidia-ml-py
fi
# Création de liens symboliques pour faciliter l'utilisation
echo "Création de liens symboliques pour les scripts..."
chmod +x chat.py
chmod +x monitor.py
# Affichage du résumé
echo ""
echo "=== Configuration terminée ==="
echo "Pour activer l'environnement virtuel, exécutez:"
echo " source llmlab/bin/activate"
echo ""
echo "Pour lancer le moniteur système:"
echo " ./monitor.py"
echo ""
echo "Pour utiliser un agent LLM:"
echo " ./chat.py [nom de l'agent]"
echo ""
echo "Agents disponibles:"
echo " cursor - CodeLlama 13B Python (pour le code)"
echo " obsidian - Llama2 13B (pour la gestion des connaissances)"
echo " test - Mistral 7B (pour les tests rapides)"
echo ""