Oldpan/Pytorch-Memory-Utils
PyTorch utility for tracking GPU memory usage of models and training operations.
Not currently ranked — collecting fresh signals.
star history
This repository provides Python scripts to monitor and calculate GPU memory consumption during PyTorch model training. It can estimate the memory footprint of individual model layers including parameters and intermediate variables for both forward and backward passes. The tool prints detailed memory breakdowns by tensor size, dtype, and class type, helping developers optimize memory usage when training deep learning models.
Frequently asked
- What is Oldpan/Pytorch-Memory-Utils?
- PyTorch utility for tracking GPU memory usage of models and training operations.
- Is Pytorch-Memory-Utils open source?
- Yes — Oldpan/Pytorch-Memory-Utils is an open-source project tracked on heatdrop.
- What language is Pytorch-Memory-Utils written in?
- Oldpan/Pytorch-Memory-Utils is primarily written in Python.
- How popular is Pytorch-Memory-Utils?
- Oldpan/Pytorch-Memory-Utils has 1k stars on GitHub.
- Where can I find Pytorch-Memory-Utils?
- Oldpan/Pytorch-Memory-Utils is on GitHub at https://github.com/Oldpan/Pytorch-Memory-Utils.