Skip to main content

A CodeCarbon wrapper for tracking and reporting energy consumption of ML experiments

Project description

LET - Lamarr Energy Tracker

Lamarr Energy Tracker logo
A simple wrapper around CodeCarbon for tracking and reporting local energy consumption from Python.

Features

  • 🧩 Simple extension to CodeCarbon software
  • 👨‍💻 Three lines of code to report on environmental impacts of your research experiments
  • 🔍 Enables smart assessment of Ground-Truth Energy Consumption
  • 📈 Integrates with STREP for resource-versus-quality comparisons
  • 💚 Help to make AI research and the Lamarr Institute more resource-aware

💻 Installation

As a Python library, you can simply install it by running

pip install lamarr-energy-tracker

⚡ Usage

LET should be used for custom compute setups (e.g., experiments running on a desktop, workstation, laptop). If you use the Lamarr Cluster, your resource consumption will be automatically tracked (more info soon), so you do not need to use LET. You can integrate LET in your Python code like this:

from lamarr_energy_tracker import EnergyTracker

# Either use as a context manager
with EnergyTracker(project_name="your_research_project") as tracker:
    # Your resource-heavy code here
    pass

# Or manually
tracker = EnergyTracker(project_name="your_research_project")
tracker.start()
# Your resource-heavy code here
tracker.stop()

Once stopped, the tracker will print the energy consumption of your experiment as well as a summary statement that you can copy into your paper, describing the environmental impact of all your performed experiments for the given project and hardware:

Using CodeCarbon 3.2.3, the energy consumption of running all experiments on an Intel(R) Core(TM) i7-10610U CPU is estimated to 0.135 kWh. This corresponds to estimated carbon emissions of 0.051 kg of CO2-equivalents, assuming a carbon intensity of 380 gCO2/kWh~\cite{lamarr_energy_tracker,codecarbon}. Note that these numbers are underestimations of actual resource consumption and do not account for overhead factors or embodied impacts~\cite{ai_energy_validation}.

Per default, the tracker stores data about tracked resource consumption in a central emissions.csv file, located in ~/.let/. You can provide a different output_dir or access the tracking results as follows (use arguments to only investigate specific projects):

from lamarr_energy_tracker import load_summary, print_paper_statement, delete_results

# access a pandas dataframe with all tracked resource data
df = load_results()
# print the summary of all tracked resource data
print_paper_statement()
# delete the centrally stored resource data
delete_results()

You can also print the statement directly from the terminal:

python -m lamarr_energy_tracker.print_paper_statement # Default arguments

python -m lamarr_energy_tracker.print_paper_statement --output_dir DIR --project_name NAME --hostname HOST # For additional filtering

❓ Assumptions and Estimation Errors

As mentioned in the impact statement above, the information obtained by CodeCarbon and LET are mere estimates of the ground-truth energy consumption. The tracking works especially well for NVIDIA GPUs (via NVML) and Linux setups, however dynamic CPU profiling with RAPL requires to run all code with sudo. If you want to run code without sudo, you can also run our RAPL access rights script before executing your code.

While the tracker assumes that all code is executed in Germany, you can also provide a different country_iso_code to change the carbon intensity constant, among some other arguments. For more information on the methodology and shortcomings of the tracker, please refer to the CodeCarbon documentation.

If you use some other energy estimation approach, such as the static Machine Learning CO2 Impact Calculator or custom profiling software like jetson-stats (for NVIDIA Jetson [Thor, Orin, Xavier, Nano, TX] series), you can also use LET to print out a custom impact statement, based on the provided methodology, hardware and energy consumption:

# from command-line
python -m lamarr_energy_tracker.print_paper_statement --methodology "the CO2 Impact Calculator" --hardware "NVIDIA GTX 1080 GPU" --consumed_energy 3.2

# from Python
from lamarr_energy_tracker import print_custom_paper_statement
print_custom_paper_statement(methodology="the CO2 Impact Calculator", hardware="NVIDIA GTX 1080 GPU", consumed_energy=3.2):

# outputs:
# Using the CO2 Impact Calculator, the energy consumption of running all experiments on an NVIDIA GTX 1080 GPU is estimated to 3.200 kWh.This corresponds to estimated carbon emissions of 1.216 kgCO2-equivalents, assuming a carbon intensity of 380 gCO2/kWh~\cite{lamarr_energy_tracker,codecarbon}. Note that these numbers are underestimations of actual resource consumption and do not account for overhead factors or embodied impacts~\cite{ai_energy_validation}.

Finally, the comparisons printed in each statement are distilled from How Bad Are Bananas? The Carbon Footprint of Everything by Mike Berners-Lee. They help gaining a better intuition for carbon intensity and can be funny, but please do not take them at face value. These numbers are very subjective and (to some degree) debatable.

🔍 Ground-Truth Energy Tracking

With Smart Sockets like the Nous A1T, it is possible to track the ground-truth energy consumption of any computer that is powered over a single power socket. This repository entails code for ground-truth tracking via a REST API offered from a simple server (we use a Raspberry Pi 5). It acts as an access point for the different smart sockets and connected hosts. Make sure to https://www.youtube.com/watch?v=9M2G2EzEXAk the smart sockets, which we did by connecting a constant power consumer (light bulb) and running the following commands:

curl IP/cm?cmnd=SaveData%201 # init
curl IP/cm?cmnd=VoltageSet%20228 # set to 228 Volt
curl IP/cm?cmnd=PowerSet%2011 # set to 11 Watt
curl IP/cm?cmnd=Status%208 # check status / alignment

After that, you can start the API server:

# on your SERVER, run via command-line
python -m lamarr_energy_tracker.ground_truth_tracking --config CONFIG_FILE

# OR Python
from lamarr_energy_tracker import GroundTruthTrackingServer
server = GroundTruthTrackingServer(CONFIG_FILE)

The CONFIG_FILE should map host names to smart socket IPs in the local network via JSON syntax, e.g.:

{
  "host1": "127.0.0.1",
  "host2": "127.0.0.2",
  "host3": "127.0.0.3",
  // ...
}

After launching the server, you need to store its IP and PORT in the LET_GT_HOST and LET_GT_PORT environment variables or /home/lamarr/.let/GT_REMOTE_CONFIG file (just call python -m lamarr_energy_tracker.ground_truth_tracking --host IP --port PORT on the client). Once properly configured, you can then perform ground-truth tracking on your machine via

# on your CLIENT (on which you execute experiments), run

from lamarr_energy_tracker import GroundTruthTracker

tracker = GroundTruthTracker()
tracker.start()
# Your resource-heavy code here
results = tracker.stop()

The results will be returned as a dictionery, comprising the start_time, timestamp, duration (in seconds) and energy_consumed (in kilowatthours). TODO: Integrate with statement printing and ~/.let/ storage.

📈 Multi-Dimensional Model Performance

You can also use LET to investigate the multi-dimensional performance of AI models, by benchmarking resource consumption and predictive quality. For that, you can for example integrate LET / CodeCarbon with MLflow and the STREP framework, allowing you to assemble and explore performance results via csv files. Examplary code can be found in the "Ground-Truthing AI Energy Consumption" repository. If you need a co-author, struggle to perform these evaluations, or want an expert opinion on your approach, feel free to reach out to Raphael.

STREP TABULUS

🤝 Collaborate

In order to become truly resource-aware, we hope to assemble impact reports about the resource consumption of research projects being conducted at Lamarr Institute. Please send your emissions.csv files to Sebastian, such that we can include your experiments in our reports. Feel free to add additional information, such as a description of the project and a link to the paper or associated code repository.

Gears icon

📝 Citing

If you use this tool to report your energy consumption, please cite the following literature:

@software{lamarr_energy_tracker,
  author = {Buschjäger, Sebastian and Fischer, Raphael},
  title  = {{Lamarr} {Energy} {Tracker}},
  year   = {2025},
  url    = {https://github.com/lamarr-institute/lamarr-energy-tracker},
}
@software{codecarbon,
  author    = {Courty, Benoît and
               Schmidt, Victor and
               Kamal, Goyal and
               others},
  title     = {mlco2/codecarbon: v3.2.3},
  year      = 2025,
  publisher = {Zenodo},
  version   = {v3.2.3},
  doi       = {10.5281/zenodo.18731928},
  url       = {https://doi.org/10.5281/zenodo.18731928},
}
@misc{ai_energy_validation,
  title  = {Ground-Truthing {AI} Energy Consumption: {Validating} {CodeCarbon} Against External Measurements}, 
  author = {Raphael Fischer},
  year   = {2025},
  doi    = {10.48550/arXiv.2509.22092},
  url    = {https://arxiv.org/abs/2509.22092}, 
}

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

Copyright (c) Resource-Aware ML Research Team @ Lamarr Institute

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

lamarr_energy_tracker-0.1.2.tar.gz (19.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

lamarr_energy_tracker-0.1.2-py3-none-any.whl (13.7 kB view details)

Uploaded Python 3

File details

Details for the file lamarr_energy_tracker-0.1.2.tar.gz.

File metadata

  • Download URL: lamarr_energy_tracker-0.1.2.tar.gz
  • Upload date:
  • Size: 19.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.0

File hashes

Hashes for lamarr_energy_tracker-0.1.2.tar.gz
Algorithm Hash digest
SHA256 1553f03605f23f6125df8cdf47a386be01fd1585121039d9ec569c48f8ad400c
MD5 515911b2cac2744262ce9472178e8666
BLAKE2b-256 c2c4cd5b38c61515499368055c04fe752d86e1c79ad8dd18470309e17020c6d1

See more details on using hashes here.

File details

Details for the file lamarr_energy_tracker-0.1.2-py3-none-any.whl.

File metadata

File hashes

Hashes for lamarr_energy_tracker-0.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 c03446a82f90bf7b92f2d5d82b3a1f3b666e77c893ddac6b17c1546a04166a92
MD5 0c2096b99830280e790cb102d4884557
BLAKE2b-256 2708fa5948b2a62934c5ffce2ebaa493213ff34cbe7d4ab74ee90be8760f2f1d

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page