mendevi.measures.gpu

Query the GPUs activity.

Classes

UsageGPU([sleep])

Use pynvml through a python context manager.

Functions

measure()

Get a instantaneous capture of all gpus.

Details

class mendevi.measures.gpu.UsageGPU(sleep: Real = 0.05)[source]

Use pynvml through a python context manager.

Examples

>>> import time
>>> from mendevi.measures.gpu import UsageGPU
>>> with UsageGPU() as gpu:
...     time.sleep(1)
...
>>>

Initialize the usage context.

Parameters

sleepfloat, default=50e-3

The time interval between 2 measures (in s).

mendevi.measures.gpu.measure() → dict[str][source]

Get a instantaneous capture of all gpus.