nntool.slurm.accelerator.utils¶

Functions¶

`nvidia_smi_gpu_memory_stats`	Parse the nvidia-smi output and extract the memory used stats.
`nvidia_smi_gpu_memory_stats_str`	Parse the nvidia-smi output and extract the memory used stats.

nntool.slurm.accelerator.utils.nvidia_smi_gpu_memory_stats()[source]¶

Parse the nvidia-smi output and extract the memory used stats.

nntool.slurm.accelerator.utils.nvidia_smi_gpu_memory_stats_str()[source]¶

Parse the nvidia-smi output and extract the memory used stats.