diff --git a/EBMs/requirements.txt b/EBMs/requirements.txt
index 18b1f22..9a6f85b 100644
--- a/EBMs/requirements.txt
+++ b/EBMs/requirements.txt
@@ -1,6 +1,6 @@
 scipy==1.10.0
 horovod==0.24.0
-torch==1.13.1
+torch==2.2.0
 torchvision==0.6.0
 six==1.11.0
 imageio==2.8.0