diff --git a/python/requirements/ml/requirements_dl.txt b/python/requirements/ml/requirements_dl.txt
index b62c2bb8ae2e..d95360732ada 100644
--- a/python/requirements/ml/requirements_dl.txt
+++ b/python/requirements/ml/requirements_dl.txt
@@ -2,7 +2,7 @@
 # For GPU Docker images, you should install requirements_ml_docker.txt afterwards.
 
 tensorflow==2.9.0
-tensorflow-probability==0.17.0
+tensorflow-probability==0.18.0
 
 # If you make changes to the torch versions below, please also make the corresponding changes to `requirements_ml_docker.txt`!