Gaiatheia is a machine learning benchmarking tool that tracks carbon emissions using CodeCarbon and integrates with Prometheus + Grafana for real-time monitoring.
- Tracks CO₂ emissions for ML models using
CodeCarbon - Pushes emissions data to
PrometheusviaPushgateway - Visualizes emissions in Grafana
First, clone the repository:
git clone https://github.com/yourusername/gaiatheia.git
cd gaiatheiaIf you’re not using Docker, install the required Python dependencies:
pip install -r requirements.txtEnsure CodeCarbon is installed:
pip install codecarbonRun the scripts to track the emissions. There are two ways to monitor the system:
- Export Data to CSV format
- Use Prometheus & Grafana to Track(currently in MVP phase)
There are 3 simple experiments attached to it, you can run those or you can use the Template.py file and follow the instruction to build and run your own model.
python experiment_1.pyTo track and visualize emissions, use Docker Compose:
docker compose upor if you are using podman
podman compose upThis will start:
- Prometheus (
http://localhost:9090) - Pushgateway (
http://localhost:9091) - Grafana (
http://localhost:3000)
- Open Grafana:
http://localhost:3000 - Login with default credentials:
Username: admin Password: admin - Add Prometheus as a Data Source (
http://prometheus:9090) - Import a new dashboard and use the metric:
codecarbon_emissions_kg
Want to improve Gaiatheia? Feel free to open an issue or submit a pull request!