diff --git a/requirements.txt b/requirements.txt
index 25111e898..6fae5c549 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-torch==1.4.0
+torch==2.2.2
 torchvision==0.5.0
 numpy==1.16.1
 matplotlib
@@ -21,4 +21,50 @@ sklearn
 umap-learn
 umap-learn[plot]
 shapely
-kivy
\ No newline at end of file
+kivy
+torchvision==0.5.0
+numpy==1.16.1
+matplotlib
+svgwrite
+cairosvg
+IPython
+six>=1.12.0
+tensorflow
+tensorboardX
+pillow
+svglib
+tqdm
+jupyter
+scikit-image
+pandas
+networkx
+moviepy
+numba
+sklearn
+umap-learn
+umap-learn[plot]
+shapely
+kivy
+torchvision==0.5.0
+numpy==1.16.1
+matplotlib
+svgwrite
+cairosvg
+IPython
+six>=1.12.0
+tensorflow
+tensorboardX
+pillow
+svglib
+tqdm
+jupyter
+scikit-image
+pandas
+networkx
+moviepy
+numba
+sklearn
+umap-learn
+umap-learn[plot]
+shapely
+kivy