diff --git a/final_project/requirements.txt b/final_project/requirements.txt
index 885142e..e9fbc08 100644
--- a/final_project/requirements.txt
+++ b/final_project/requirements.txt
@@ -5,5 +5,5 @@ nvidia-ml-py3==7.352.0
 psutil==5.7.0
 six==1.14.0
 termcolor==1.1.0
-torch==1.4.0
+torch==2.2.0
 tqdm==4.44.1