optimize_memory.py

"""
Script opcional para optimizar la memoria antes de ejecutar la aplicación.
Ejecuta este script antes de iniciar la aplicación si experimentas problemas de memoria.
"""

import gc
import os
import torch
import psutil

def optimize_memory():
    print("Optimizando memoria del sistema...")
    
    # Forzar recolección de basura de Python
    gc.collect()
    
    # Liberar caché de CUDA si está disponible
    if torch.cuda.is_available():
        torch.cuda.empty_cache()
        print("Caché de CUDA liberada")
    
    # Información sobre memoria disponible
    memory = psutil.virtual_memory()
    print(f"Memoria total: {memory.total / (1024**3):.2f} GB")
    print(f"Memoria disponible: {memory.available / (1024**3):.2f} GB")
    print(f"Porcentaje usado: {memory.percent}%")
    
    # Configurar variables de entorno para reducir uso de memoria
    os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"
    os.environ["TOKENIZERS_PARALLELISM"] = "false"
    
    print("Optimización de memoria completada")

if __name__ == "__main__":
    optimize_memory()