megengine.get_max_allocated_memory

get_max_allocated_memory(device=None)

Returns the maximum memory occupied by tensors on the computing device in bytes.

Due to the asynchronous execution of MegEngine, please call megengine._full_sync before calling this function in order to get accurate value.