diff --git a/setup.py b/setup.py
index 1873a22..93e2323 100644
--- a/setup.py
+++ b/setup.py
@@ -32,7 +32,7 @@
 		"numpy == 1.22.0",
 		"pandas == 1.3.3",
 		"matplotlib==3.4.3",
-		 "torch==1.9.0",
+		 "torch==2.2.0",
         "xgboost==1.4.2",
 		"tqdm==4.62.2"