torch.cuda.memory.memory_summary#

torch.cuda.memory.memory_summary(device=None,abbreviated=False)[source]#

Return a human-readable printout of the current memory allocator statistics for a given device.

This can be useful to display periodically during training, or whenhandling out-of-memory exceptions.

Parameters

• device (torch.device orint,optional) – selected device. Returnsprintout for the current device, given bycurrent_device(),ifdevice isNone (default).

• abbreviated (bool,optional) – whether to return an abbreviated summary(default: False).

Return type

str

Note

SeeMemory management for more details about GPU memorymanagement.