diff --git a/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt b/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
index ff39e67..f9c57ae 100644
--- a/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
+++ b/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
@@ -1,5 +1,5 @@
 torch==2.7.0
 torchvision==0.22.0
 # the below are used by AML, can be deleted if not using AML
-mlflow==2.21.3
+mlflow==3.1.0
 azureml-mlflow==1.60.0