Shortcuts

torch.cuda.list_gpu_processes

torch.cuda.list_gpu_processes(device=None)[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

str

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