blazingsql.BlazingContext.get_max_memory_used¶
-
BlazingContext.get_max_memory_used()¶ This function returns a dictionary which contains as key the gpuID and as value the max memory (bytes)
# single-GPU >>> from blazingsql import BlazingContext >>> bc = BlazingContext() >>> max_mem_used = bc.get_max_memory_used() >>> print(max_mem_used)
{0: 4234220154}
# multi-GPU (4 GPUs): >>> from blazingsql import BlazingContext >>> from dask_cuda import LocalCUDACluster >>> from dask.distributed import Client >>> cluster = LocalCUDACluster() >>> client = Client(cluster) >>> bc = BlazingContext(dask_client=client, network_interface=’lo’) >>> max_mem_used = bc.get_max_memory_used() >>> print(max_mem_used)
- {0: 4234220154, 1: 4104210987,
2: 4197720291, 3: 3934320116}