perun.backend.nvml

Nvidia Mangement Library Source definition.

Attributes

log

Classes

NVMLBackend

NVMLSource class.

Module Contents

perun.backend.nvml.log

class perun.backend.nvml.NVMLBackend

Bases: perun.backend.backend.Backend

NVMLSource class.

Setups connection to NVML and creates relevant devices

id = 'nvlm'

name = 'NVIDIA ML'

description: str = 'Access GPU information from NVML python bindings'

setup() → None

Init pynvml and gather number of devices.

close() → None

Backend shutdown code.

availableSensors() → dict[str, tuple]

Return a dictionary with all available sensors.

Each entry contains the backend id and type of sensor.

Returns:

Dictionary with device ids and measurement unit.

Return type:

dict[str, tuple]

getSensors(deviceList: set[str]) → list[perun.data_model.sensor.Sensor]

Gather sensor object based on a set of device ids.

Parameters:

deviceList (set[str]) – Set containing divice ids.

Returns:

List with Sensor objects.

Return type:

list[Sensor]