GPUtil is a Python module for getting the GPU status from NVIDIA GPUs
using nvidia-smi.
