pynvml>=11.4.1
numpy>=1.16.0
