mxnet.npx.gpu_memory_info

gpu_memory_info(device_id=0)

Query CUDA for the free and total bytes of GPU global memory.

Parameters

device_id (int, optional) – The device id of the GPU device.

Raises

Will raise an exception on any CUDA error. –

Returns

(free, total)

Return type

(int, int)