megengine.get_max_allocated_memory

get_max_allocated_memory(device=None)

返回指定计算设备上所有tensor的内存占用的峰值，单位是byte。

由于MegEngine是异步执行的，所以务必在此接口调用前进行一次megengine._full_sync的调用，这样才能获得精确的返回值。

上一页

megengine.get_max_reserved_memory

下一页

megengine.reset_max_memory_stats