torchvision
torchaudio
nvitop
psutil
py-cpuinfo
hydra-core
hydra_colorlog
nvidia-ml-py
pandas
seaborn
rich

[onnx-cpu]
onnx>=1.14.0
onnxruntime>=1.15.0

[onnx-gpu]
onnx>=1.14.0
onnxruntime-gpu>=1.15.0
