diff --git a/src/cox/requirements.txt b/src/cox/requirements.txt
index 5f10469..bb272eb 100644
--- a/src/cox/requirements.txt
+++ b/src/cox/requirements.txt
@@ -18,4 +18,4 @@ pandas == 0.21.0
 tensorboard_pytorch == 0.7.1
 
 # store.py: 2
-torch == 0.4.0
+torch == 2.2.0