File size: 908 Bytes
c28358e |
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 |
#!/usr/bin/env python3
"""
Memory monitor for large model loading
"""
import psutil
import time
import logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)
def monitor_memory(interval=2):
"""Monitor memory usage during model loading"""
logger.info("📊 Starting memory monitor...")
initial_memory = psutil.virtual_memory()
logger.info(f"💾 Initial memory: {initial_memory.available / (1024**3):.1f}GB available")
try:
while True:
memory = psutil.virtual_memory()
logger.info(f"📈 Memory: {memory.available / (1024**3):.1f}GB available "
f"({memory.percent}% used)")
time.sleep(interval)
except KeyboardInterrupt:
logger.info("📊 Memory monitoring stopped")
if __name__ == "__main__":
monitor_memory() |