tomodrgn.utils.check_memory_usage

check_memory_usage(device: device) → list[str]

Get the current memory usage of the specified device_type.

Parameters:

device – the (class of) hardware device for which to report memory usage. Allowable device types include cuda and cpu

Returns:

List of VRAM or RAM usage of each visible instance of the specified device_type.