diff --git a/python/requirements.txt b/python/requirements.txt
index c15d8e7..c304006 100644
--- a/python/requirements.txt
+++ b/python/requirements.txt
@@ -8,7 +8,7 @@ docopt
 pyyaml
 protobuf
 grpcio
-torch==1.4
+torch==2.2.0
 pandas
 scipy
 ipykernel