neuronbox-runtime 0.1.0

Local ML runner: declarative neuron.yaml, model store, daemon, and terminal dashboard
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

//! Single host probe layer: GPU, platform, likely PyTorch backend.

mod apple;
mod nvidia;
#[cfg(all(target_os = "linux", feature = "nvml"))]
mod nvml_linux;
mod probe;
mod rocm;
mod snapshot;

pub use nvidia::{compute_apps_display_lines, compute_apps_pid_memory_mb};
pub use probe::HostProbe;
pub use snapshot::{
    GpuRecord, HostSnapshot, PlatformInfo, ProbeStatus, TrainingBackend,
    HOST_SNAPSHOT_SCHEMA_VERSION,
};