torch.cuda.memory_usage — PyTorch 2.7 documentation (original) (raw)

torch.cuda.memory_usage(device=None)[source][source]¶

Return the percent of time over the past sample period during which global (device) memory was being read or written as given by nvidia-smi.

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

Warning: Each sample period may be between 1 second and 1/6 second, depending on the product being queried.

• torch.cuda.memory_usage
  • memory_usage()

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.