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.

Returns:

(free, total) : (int, int)

Raises:

Will raise an exception on any CUDA error.