Shortcuts

mmengine.device.get_max_cuda_memory

mmengine.device.get_max_cuda_memory(device=None)[source]

Returns the maximum GPU memory occupied by tensors in megabytes (MB) for a given device. By default, this returns the peak allocated memory since the beginning of this program.

Parameters:

device (torch.device, optional) – selected device. Returns statistic for the current device, given by current_device(), if device is None. Defaults to None.

Returns:

The maximum GPU memory occupied by tensors in megabytes for a given device.

Return type:

int

Read the Docs v: v0.10.4
Versions
latest
stable
v0.10.4
v0.10.3
v0.10.2
v0.10.1
v0.10.0
v0.9.1
v0.9.0
v0.8.5
v0.8.4
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.