torch>=2.0
psutil
numpy
