Clear cached memory instances and free GPU memory
()
| 612 | |
| 613 | |
| 614 | def clear_memory_cache(): |
| 615 | """Clear cached memory instances and free GPU memory""" |
| 616 | global _memory_cache |
| 617 | |
| 618 | num_entries = len(_memory_cache) |
| 619 | _memory_cache.clear() |
| 620 | |
| 621 | # Clear PyTorch GPU cache |
| 622 | try: |
| 623 | import torch |
| 624 | if torch.cuda.is_available(): |
| 625 | torch.cuda.empty_cache() |
| 626 | logger.info(f"Cleared memory cache ({num_entries} entries) and GPU cache") |
| 627 | else: |
| 628 | logger.info(f"Cleared memory cache ({num_entries} entries)") |
| 629 | except ImportError: |
| 630 | logger.info(f"Cleared memory cache ({num_entries} entries)") |