torch.cuda.memory\_summary ========================== .. currentmodule:: torch.cuda .. autofunction:: memory_summary