torch>=1.6
torchvision
numpy
matplotlib
psutil
PyGLM
