diff --git a/environment.yaml b/environment.yaml
index 87565b0..1d133f4 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -44,7 +44,7 @@ dependencies:
   - sphinx=5.*
   - tokenizers=0.*
   - torchaudio=2.*
-  - torchmetrics=0.*
+  - torchmetrics=1.*
   - tqdm=4.*
   - transformers=4.*
   - wandb=0.*
diff --git a/requirements.txt b/requirements.txt
index b41e60e..dce5679 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -26,7 +26,7 @@ tokenizers==0.13.2
 torch==2.0.1
 torchaudio==2.0.2
 torch-geometric==2.3.0
-torchmetrics==0.11.4
+torchmetrics==1.3.1
 torchvision==0.15.2
 tqdm==4.65.0
 transformers==4.36.0
diff --git a/setup.py b/setup.py
index 2476cf4..5563116 100644
--- a/setup.py
+++ b/setup.py
@@ -29,7 +29,7 @@
         "seaborn==0.12.2",
         "sphinx==5.0.2",
         "tokenizers==0.13.2",
-        "torchmetrics==0.11.4",
+        "torchmetrics==1.3.1",
         "tqdm==4.65.0",
         "transformers==4.36.0",
         "wandb==0.15.3",