pynvml>=11.4.1
numpy>=1.16.0
torch>=1.0.0
