Shortcuts

mmengine.device.get_max_cuda_memory

mmengine.device.get_max_cuda_memory(device=None)[源代码]

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.

参数

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

返回

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

返回类型

int

Read the Docs v: latest
Versions
latest
stable
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.