diff --git a/tests/data/serve_resources/mlflow/pytorch/conda.yaml b/tests/data/serve_resources/mlflow/pytorch/conda.yaml
index be61456197..beecdbab08 100644
--- a/tests/data/serve_resources/mlflow/pytorch/conda.yaml
+++ b/tests/data/serve_resources/mlflow/pytorch/conda.yaml
@@ -17,8 +17,8 @@ dependencies:
   - pandas==2.2.1
   - pyyaml==6.0.1
   - requests==2.31.0
-  - torch==2.0.1
-  - torchvision==0.15.2
+  - torch>=2.6.0
+  - torchvision>=0.17.0
   - tqdm==4.66.2
   - scikit-learn==1.3.2
 name: mlflow-env
diff --git a/tests/data/serve_resources/mlflow/pytorch/requirements.txt b/tests/data/serve_resources/mlflow/pytorch/requirements.txt
index 0446ed5053..450bcbfada 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.2.0
-torchvision==0.17.0
+torch>=2.6.0
+torchvision>=0.17.0
 tqdm==4.66.3
diff --git a/tests/unit/sagemaker/jumpstart/constants.py b/tests/unit/sagemaker/jumpstart/constants.py
index 0c9065feb5..83e8a44a32 100644
--- a/tests/unit/sagemaker/jumpstart/constants.py
+++ b/tests/unit/sagemaker/jumpstart/constants.py
@@ -17393,7 +17393,7 @@
             "texttable==1.6.7",
             "tokenize-rt==5.1.0",
             "tokenizers==0.13.3",
-            "torch==2.2.0",
+            "torch>=2.6.0",
             "transformers==4.33.3",
             "triton==2.2.0",
             "typing-extensions==4.8.0",