megengine.get_allocated_memory

get_allocated_memory(device=None)

返回指定的计算设备上tensor占用的内存总量，单位是byte。

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