torch.cuda.list_gpu_processes — PyTorch 2.7 documentation (original) (raw)
torch.cuda.list_gpu_processes(device=None)[source][source]¶
Return a human-readable printout of the running processes and their GPU memory use for a given device.
This can be useful to display periodically during training, or when handling out-of-memory exceptions.
Parameters
device (torch.device or int, optional) – selected device. Returns printout for the current device, given by current_device(), if device is None (default).
Return type
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.
