ml-energy / zeus

Deep Learning Energy Measurement and Optimization
https://ml.energy/zeus
Apache License 2.0
193 stars 25 forks source link

Support for NVIDIA Jetson platforms #103

Open jaywonchung opened 1 month ago

jaywonchung commented 1 month ago

NVML is not supported on Jetson platforms, e.g. Jetson Nano. Supporting Jetson platforms can be very useful for people who do ML/MLSys research on embedded platforms. Related discussion entry: https://github.com/ml-energy/zeus/discussions/102

We can make this a new zeus.device.

Things to figure out:

jaywonchung commented 1 month ago

Ref: https://github.com/alumet-dev/alumet