diff --git a/requirements.txt b/requirements.txt
index 4c4317b..65f57a5 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -14,7 +14,7 @@ scikit-learn >= 1.0
 networkx >= 2.4
 pydot >= 1.4.1
 tabulate >= 0.8.6
-pytorch >= 1.3.1
+torch >= 1.3.1
 scikit-hep >= 0.5.1
 bokeh >= 1.4.0
 psutil >= 5.6.7