torch
numpy
psutil>=5.9
