emt 0.1.0a1

A tool for granular energy measurement, empowering digital sustainability in machine learning and AI workflows.

Energy Monitoring Tool (EMT)

Track and analyze energy usage of your software application(s) — lightweight, reliable and effortless to integrate.

EMT is a lightweight tool capable of tracking and reporting the energy consumption of software systems with process-level granularity. While particularly valuable for compute-intensive workloads like machine learning, it's designed for broad applicability across various use cases. Our mission is to simplify and standardize monitoring and reporting of the energy usage of the the digital realm. By making it visible and accessible, EMT helps teams to reduce their environmental impact and advances digital sustainability.

🚀 Features

• Real-time energy utilization tracking.
• Device-level breakdown of energy consumption.
• Energy/Power attribution to a process of interest in a multi-process shared resource setting.
• Modular and extendable software architecture, currently supports following powergroups:
  • CPU(s) with RAPL capabilities.
  • Nvidia GPUs.
• Visualization interface for energy data using TensorBoard, making it easy to analyze energy usage trends.

Supported Platforms & Hardware

• Linux
• Hardware
  • Nvidia GPU through NVML
  • Intel & AMD x86 sockets through RAPL

Road Map

• Environmentally conscious coding tips.
• Virtual CPU(s) covered by Teads dataset.
• Add support for Windows through PCM/OpenHardwareMonitor
• Extend harware support

🌍 Why EMT?

In the era of climate awareness, it's essential for developers to contribute to a sustainable future. EMT Tool empowers you to make informed decisions about your code's impact on the environment and take steps towards writing more energy-efficient software.

🛠️ Getting Started

Install the latest EMT package from the Python Package Index (PyPI):

pip install emt

# verify installation and the version
python -m emt --version

# One time configuration to allow access to the RAPL files
emt_cfgup

Usage

The tool supports two usage modes:

• Python Context Manager
  Fully implemented and ideal for instrumenting Python code directly. This mode allows developers to wrap specific code blocks to measure energy consumption with precision.
• Command-Line Interface (CLI)
  Designed to tag and monitor running application without modifying the code.
  This mode is currently under active development and will be available soon.

Using Python Context Managers

import torch
from emt import EnergyMonitor

# Dummy function
def add_tensors_gpu():
    device = torch.device(device if torch.cuda.is_available() else "cpu")
    # Generate random data
    a = torch.randint(1, 100, (1000,), dtype=torch.int32, device=device)
    b = torch.randint(1, 100, (1000,), dtype=torch.int32, device=device)

    return a + b

# Create a context manager
with EnergyMonitor() as monitor:
    add_tensors_gpu()

print(f"energy consumption: {monitor.consumed_energy}")

Refer to the following folder for example codes: 📁 examples/

Dynamic Child Processes

In some cases, such as when using non-Python applications within a script (s. example below) or when workers are spawned dynamically, the child processes are not be created before the EnergyMonitor, which must therefore reload the child processes.
This can be enabled through the environment variable EMT_RELOAD_PROCS.

import json, os

# Enforce reloading (child) processes
os.environ["EMT_RELOAD_PROCS"] = "1"
from emt import EnergyMonitor

with EnergyMonitor() as monitor:
   # Example: Run non-Python GPU binary
   cmd = "lmp -in input.in -sf gpu"
   os.system(cmd)

print(f"energy consumption: {monitor.consumed_energy}")

⚙️ Methodology

The EMT context manager spawns a separate thread to monitor energy usage for CPUs and GPUs at regular intervals. It also tracks the utilization of these resources by the monitored process. EMT then estimates the process's share of the total energy consumption by proportionally assigning energy usage based on the resource utilization of the process.

📧 Contact

For any inquiries or discussions, feel free to reach out to us:
Rameez Ismail: [e-mail]
Sophie Thornander: [e-mail]
Arlette van Wissen: [e-mail]

Let's code responsibly and make a positive impact on the environment! 🌍✨

Acknowledgment

This project was originally initiated at Philips and continues to be actively maintained by the Philips Responsible AI team. We extend our sincere gratitude to all current and former contributors at Philips whose expertise, vision, and commitment to sustainability have been instrumental in shaping the Energy Monitoring Tool (EMT).