viznvsmi 0.0.1

NVSMI plugins for VizTracer

VizNvsmi

VizTracer plugins for NVSMI. It is inspired by the official VizTracer plugings: vizplugins.

Features

VizNvsmi currently only supports NVIDIA metrics tracking via nvidia-smi.

Install

The preferred way to install VizNvsmi is via pip:

pip install viznvsmi

Basic Usage

VizNvsmi should be used with VizTracer.

You can use VizNvsmi and the plugin via command line:

viztracer --plugin viznvsmi.utilization.memory -i 0 -- my_script.py arg1 arg2

Or equivalent syntax:

viztracer --plugin "viznvsmi --utilization.memory -i 0" -- my_script.py arg1 arg2

You can also add the plugin to VizTracer programmatically as described in the official documentation.

Finally, VizNvsmi provides a PyTorch Lightning Callback that can be used as part of model training.