diff --git a/tests/e2e/mnist_pip_requirements.txt b/tests/e2e/mnist_pip_requirements.txt
index 05ed5816..81632666 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.5.1
 torchvision==0.19.0
 minio