Prune down Docker image size

ml-energy / zeus

Deep Learning Energy Measurement and Optimization

https://ml.energy/zeus

Apache License 2.0

180 stars 24 forks source link

Prune down Docker image size #18

Closed jaywonchung closed 10 months ago

jaywonchung commented 10 months ago

Our Docker image used to build on top of the dev variant of NVIDIA/CUDA images in order to be able to build the C++ zeus_monitor (Actually we didn't even have to do this by using multi-stage builds and copying over just the binary, but whatever.)

Now that we're pure Python, switch to a lighter image and test stuff.

jaywonchung commented 10 months ago

Implemented by f42ff2c8134ff00165e1b6a91a1ed7e507f7b023. Image size is down by 3x now!