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