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: stable
Versions
latest
stable
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
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.