torch>=1.3
graphviz
