ml-energy / zeus

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

Support for NVIDIA Jetson platforms #103

Open jaywonchung opened 4 months ago

jaywonchung commented 4 months 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 3 months ago

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