[Feature Request] Utility API for querying GPU memory

apache / mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more

https://mxnet.apache.org

Apache License 2.0

20.78k stars 6.79k forks source link

[Feature Request] Utility API for querying GPU memory #12126

Open sandeep-krishnamurthy opened 6 years ago

sandeep-krishnamurthy commented 6 years ago

Platform independent API (without using nvidia-smi) for querying GPU memory. C implementation is provided by @sbodenstein in this PR - https://github.com/apache/incubator-mxnet/pull/12083#pullrequestreview-145347358

ddavydenko commented 5 years ago

I would also suggest to extend the C implementation to support wider GPU info. More precisely I suggest to add number of GPUs available into output of this method. And then, wrap it all up and expose via Python API.