diff --git a/tests/data/serve_resources/mlflow/pytorch/requirements.txt b/tests/data/serve_resources/mlflow/pytorch/requirements.txt
index de976327a4..0446ed5053 100644
--- a/tests/data/serve_resources/mlflow/pytorch/requirements.txt
+++ b/tests/data/serve_resources/mlflow/pytorch/requirements.txt
@@ -11,6 +11,6 @@ packaging==21.3
 pandas==2.2.1
 pyyaml==6.0.1
 requests==2.32.2
-torch==2.0.1
-torchvision==0.15.2
+torch==2.2.0
+torchvision==0.17.0
 tqdm==4.66.3