nvtop 0.1.1

nvtop: An NVIDIA SMI'esk GPU Monitoring tool for your terminal.
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use std::time::Duration;

use anyhow::Result;
use clap::Parser;
use log::{error, trace};
use nvml_wrapper::{
    enum_wrappers::device::{Clock, ClockId},
    Nvml,
};

use nvtop::{app::run, errors::NvTopError, gpu::GpuInfo, nvtop_args, termite::setup_logger};

fn main() -> Result<(), NvTopError> {
    let args = nvtop_args::Cli::parse();

    // If they've used the --log arg we write all logs to disk.
    if args.log.is_some() {
        setup_logger(args.log)?;
    } else {
        pretty_env_logger::init(); // If they've got RUST_LOG=trace on the TUI is ruined.
    }

    // Init the GPU management-layer
    let nvml = Nvml::init()?;
    trace!("Nvml init success");

    // Get the first `Device` (GPU) in the system
    let device = nvml.device_by_index(0)?;
    trace!("Compatible GPU found at [0]");

    // Do some setup for things that will _not_ change, i.e driver version etc.
    let card_type = format!("{:?}", device.brand()?);
    let driver_version = device.nvml().sys_driver_version()?;
    let cuda_version = device.nvml().sys_cuda_driver_version()? as f32;

    let misc = format!(
        "Card: {:?}    Driver Version: {}    CUDA Version: {}",
        card_type,
        driver_version,
        cuda_version / 1000.0
    );
    trace!("Setting misc = {misc}");

    dbg!(
        device.max_clock_info(Clock::Graphics)?,
        device.max_clock_info(Clock::Video)?,
        device.max_clock_info(Clock::SM)?,
        device.max_clock_info(Clock::Memory)?,
    );

    let gpu = GpuInfo {
        inner: &device,
        max_memory_clock: device.max_clock_info(Clock::Memory)?,
        max_core_clock: device.max_clock_info(Clock::Graphics)?,
        card_type,
        driver_version,
        cuda_version,
        misc,
        num_cores: device.num_cores()?,
    };
    // panic!("Starting TUI with GPU = {gpu}\n");

    if let Err(e) = run(&gpu, Duration::from_millis(args.delay)) {
        error!("app::run() -> {e}");
    }

    Ok(())
}