gputil
A Python module for programmatically retrieving NVIDIA GPU status and selecting available GPUs based on memory and load.
What is gputil?
GPUtil is a Python module that provides programmatic access to NVIDIA GPU status via the `nvidia-smi` command-line tool. It helps developers identify available GPUs based on current memory and load thresholds, automating GPU selection for tasks like deep learning training and computational workloads. The module simplifies GPU resource management in multi-GPU setups without requiring external dependencies.
Developers and researchers working with NVIDIA GPUs in Python, especially those in deep learning, scientific computing, or any field requiring efficient GPU resource allocation. It is particularly useful for teams managing shared GPU servers or running multi-GPU experiments.
GPUtil offers a straightforward, dependency-free way to interact with NVIDIA GPUs, eliminating the need to manually parse `nvidia-smi` output. Its flexible filtering and ordering options make it ideal for automating GPU selection, saving time and reducing errors in GPU-intensive workflows.
Overview
A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python
Use Cases
Best For
- Automatically selecting idle GPUs for deep learning training jobs
Related Projects
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
