Skip to main content

EcoML – Track, analyse and reduce ML carbon emissions inside Jupyter notebooks.

Project description

EcoML 🌿

Track, analyse and reduce the carbon footprint of your ML experiments — directly inside Jupyter.

PyPI version Python License: MIT


Install

pip install ecoml

To also enable Gemini AI tips (recommended):

pip install "ecoml[gpu]"   # + WMI GPU temp on Windows

Quick Start

import sys, pathlib
ROOT = pathlib.Path().resolve().parent
sys.path.insert(0, str(ROOT))

from ecoml import EcoTracker, CellHook, RecommendationEngine, GeminiAdvisor

tracker    = EcoTracker(log_path=str(ROOT / "data" / "emissions_log.csv"))
gemini     = GeminiAdvisor()                   # needs GEMINI_API_KEY in .env
recommender = RecommendationEngine(gemini=gemini)

hook = CellHook(tracker, recommender)
hook.register()
print("🌿 EcoML Ready")

After hook.register() every subsequent cell automatically shows:

  • 🌿 CO₂ produced (grams)
  • ⚙️ Hardware recommendation with confidence score
  • 💡 AI-powered tips via Gemini
  • 📊 Eco Score (0–100)

What's New in v0.2.0

Area Improvement
Recommender 12 intelligent rules (RAM, vRAM, temps, runtime, CO₂, GPU/CPU bottlenecks)
Confidence Score Dynamic 0–1 score based on how many issues are detected
RAM Monitoring System RAM % and GPU VRAM % tracked per cell
Gemini Prompts Rich context-aware prompts with all metrics for targeted advice
Eco Score Physics-inspired formula penalising load, heat, runtime & carbon
UI RAM/VRAM rows, colour-coded confidence badge, both CPU + GPU temps

Environment Setup

Create a .env file in your project root:

GEMINI_API_KEY=your_key_here

Get a free key at aistudio.google.com.


Dependencies

Automatically installed with pip install ecoml:

  • psutil — CPU / RAM metrics
  • GPUtil — GPU utilisation & temperature
  • pandas — CSV logging
  • scikit-learn — ML-based hardware predictor
  • ipython — Jupyter cell hooks
  • python-dotenv.env key loading
  • google-generativeai — Gemini AI advisor

License

MIT © EcoML Team

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

ecoml-0.2.9.tar.gz (24.6 kB view details)

Uploaded Source

Built Distribution

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

ecoml-0.2.9-py3-none-any.whl (25.9 kB view details)

Uploaded Python 3

File details

Details for the file ecoml-0.2.9.tar.gz.

File metadata

  • Download URL: ecoml-0.2.9.tar.gz
  • Upload date:
  • Size: 24.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.0

File hashes

Hashes for ecoml-0.2.9.tar.gz
Algorithm Hash digest
SHA256 958db4229c00edaef814b05832588e333d1b6601c73e847000ee532f96d014ea
MD5 32fe8de15d4f8b18aa4478d270eadea5
BLAKE2b-256 2b49ef1f71cfb52c399ac5f657d95b120437c753bc6ac706147808a7367727c2

See more details on using hashes here.

File details

Details for the file ecoml-0.2.9-py3-none-any.whl.

File metadata

  • Download URL: ecoml-0.2.9-py3-none-any.whl
  • Upload date:
  • Size: 25.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.0

File hashes

Hashes for ecoml-0.2.9-py3-none-any.whl
Algorithm Hash digest
SHA256 944bbbfdf2a102583b47e909dc93868c50177d91521af089cc67a9a0f4e58137
MD5 1f7ee475a844b3312c9540dc274d86bd
BLAKE2b-256 3c075ba2473edda4f3de86e510d2773ea9ff9450c9a6ac2ce9dd7aa7fbdfe6f4

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