Skip to content

Python3 context manager to log power consumption of ML-pipelines running on Nvidia-GPUs.

License

Notifications You must be signed in to change notification settings

done-my-way/PowerWatcher

Repository files navigation

PowerWatcher

Python3 context manager to log power consumption of an ML-pipeline running on a Nvidia-GPU.

Installation

pip install power-consumption

Usage

from power_watcher import PowerWatcher


with PowerWatcher() as pw:
    ...
    your pipeline
    ...

total_consumption = pw.total  # total power consumption

Repo

https://github.com/WGussev/PowerWatcher

About

Python3 context manager to log power consumption of ML-pipelines running on Nvidia-GPUs.

Topics

Resources

License

Stars

Watchers

Forks

Languages