diff --git a/requirements.txt b/requirements.txt
index d77a6f5..18ba222 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ tqdm
 scikit-learn
 pandas
 numpy
-torch==2.2.0
+torch==2.6.0
 torchvision==0.6
 wandb
 h5py