Create optimize_memory.py

optimize_memory.py

@@ -0,0 +1,36 @@
+"""
+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()
+