perun.backend.nvml ================== .. py:module:: perun.backend.nvml .. autoapi-nested-parse:: Nvidia Mangement Library Source definition. Attributes ---------- .. autoapisummary:: perun.backend.nvml.log Classes ------- .. autoapisummary:: perun.backend.nvml.NVMLBackend Module Contents --------------- .. py:data:: log .. py:class:: NVMLBackend Bases: :py:obj:`perun.backend.backend.Backend` NVMLSource class. Setups connection to NVML and creates relevant devices .. py:attribute:: id :value: 'nvlm' .. py:attribute:: name :value: 'NVIDIA ML' .. py:attribute:: description :type: str :value: 'Access GPU information from NVML python bindings' .. py:method:: setup() -> None Init pynvml and gather number of devices. .. py:method:: close() -> None Backend shutdown code. .. py:method:: availableSensors() -> Dict[str, Tuple] Return string ids of visible devices. :returns: Set with sensor ids. :rtype: Set[str] .. py:method:: getSensors(deviceList: Set[str]) -> List[perun.data_model.sensor.Sensor] Gather sensor object based on a set of device ids. :param deviceList: Set containing divice ids. :type deviceList: Set[str] :returns: List with Sensor objects. :rtype: List[Sensor]