gputil

1 · · Sept. 6, 2021, 9:28 a.m.
gputil A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python GPUtil is a Python module for getting the GPU status from NVIDA GPUs using nvidia-smi. GPUtil locates all GPUs on the computer, determines their availablity and returns a ordered list of available GPUs. Availablity is based upon the current memory consumption and load of each GPU. The module is written with GPU selection for Deep Learning in mind, but it is not task/library specific and it...