ml-energy / zeus

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

`ZeusProfilingService` for profiling energy with NVML energy query method on arbitrary profile windows #13

Closed Rosie-m closed 1 year ago

Rosie-m commented 1 year ago

In response to #12, this PR uses nvmlDeviceGetTotalEnergyConsumption for GPU architectures Volta and later.

Main Changes and Benefits

jaywonchung commented 1 year ago

@Rosie-m The changes look all good! Thanks a lot.

I pushed some small changes. Can you take a look?