diff --git a/tests/e2e/mnist_pip_requirements.txt b/tests/e2e/mnist_pip_requirements.txt
index 60811f18a..5555f2c5d 100644
--- a/tests/e2e/mnist_pip_requirements.txt
+++ b/tests/e2e/mnist_pip_requirements.txt
@@ -1,4 +1,4 @@
 pytorch_lightning==1.9.5
-torchmetrics==0.9.1
+torchmetrics==1.6.0
 torchvision==0.20.1
 minio