mltop 0.6.0

Resource monitor for ML engineers written in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use nvml_wrapper::{error::NvmlError, Nvml};

use nvml_wrapper::enum_wrappers::device::TemperatureSensor;

#[derive(Clone)]
pub struct GpuSnapshot {
    pub name: String,
    pub temperature: u32,
    pub max_memory: u64,  // bytes
    pub used_memory: u64, //bytes
    pub utilization: u32,
    pub max_power: u32,
    pub power_usage: u32,
    pub fan_speed: u32,
}

impl GpuSnapshot {
    pub fn from_nvml(nvml: &Nvml) -> Result<GpuSnapshot, NvmlError> {
        let device = nvml.device_by_index(0)?;
        let memory_info = device.memory_info()?;

        Ok(GpuSnapshot {
            name: device.name()?,
            temperature: device.temperature(TemperatureSensor::Gpu)?,
            max_memory: memory_info.total,
            used_memory: memory_info.used,
            utilization: device.utilization_rates()?.gpu,
            max_power: device.power_management_limit()?,
            power_usage: device.power_usage()?,
            fan_speed: device.fan_speed(0)?,
        })
    }
}