numpy
pyctlib
torch>=1.7.0
pynvml
