torch.cuda.mem_get_info
torch.cuda.mem_get_info(device=None)
Returns the global free and total GPU memory occupied for a given
   device using cudaMemGetInfo.
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:
      Tuple[int, int]
Note:
 See Memory management for more details about GPU memory
 management.
