PySide6
torch
torchvision
numpy
Pillow
matplotlib
torchmetrics
termcolor

[tensorrt]
torch-tensorrt>=2.3.0
