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

Free document hosting provided by Read the Docs.