torch.cuda.memory\_usage ======================== .. currentmodule:: torch.cuda .. autofunction:: memory_usage