megengine.get_allocated_memory

get_allocated_memory(device=None)[源代码]

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

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