diff --git a/requirements.txt b/requirements.txt
index 9889d0c..c45f3ca 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -16,7 +16,7 @@ dinov2==0.0.1.dev0
 xformers==0.0.18
 
 # --------- pytorch --------- #
-torch==2.0.0
+torch==2.3.1
 torchvision==0.15.1
 lightning @ git+https://github.com/Lightning-AI/lightning@1307b605e858560a1b93bfbbbbe039e5beb9e977#egg=lightning
 torchmetrics==0.11.4