torch>=1.9
torchvision
prettytable
openpyxl
graphviz
torchview
