Spaces:
Running
Running
File size: 1,090 Bytes
3e9bc78 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
"""
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()
|