A (user-)friendly wrapper to nvidia-smi
.
It can be used to filter the GPUs based on resource usage (e.g. to choose the least utilized GPU on a multi-GPU system).
nvsmi --help
nvsmi ls --help
nvsmi ps --help
import nvsmi
nvidiapy.get_gpus()
nvidiapy.get_available_gpus()
nvidiapy.get_gpu_processes()
- An nvidia GPU
nvidia-smi
- Python 2.7 or 3.6+
pip install --user nvsmi