Table of Contents
Shortcuts

# torch.cuda.max_memory_reserved

torch.cuda.max_memory_reserved(device=None)

  Returns the maximum GPU memory managed by the caching allocator in bytes for a given device.

  By default, this returns the peak cached memory since the beginning of this program. `reset_peak_memory_stats()` can be used to reset the starting point in tracking this metric. For example, these two functions can measure the peak cached memory amount of each iteration in a training loop.

  Parameters:

    device (torch.device or int, optional) – selected device. Returns statistic for the current device, given by `current_device()`, if `device` is `None` (default).

  Return type:

    int

  Note

  See Memory management for more details about GPU memory management.

## Docs

Access comprehensive developer documentation for PyTorch

View Docs

## Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

## Resources

Find development resources and get your questions answered

View Resources

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.