diff --git a/python/requirements/ml/requirements_dl.txt b/python/requirements/ml/requirements_dl.txt
index 04d41897a8be..131b31eab229 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.6.2
-tensorflow-probability==0.14.1
+tensorflow-probability==0.17.0
 
 # If you make changes to the torch versions below, please also make the corresponding changes to `requirements_ml_docker.txt`!