diff --git a/python/requirements/ml/requirements_dl.txt b/python/requirements/ml/requirements_dl.txt index f64b332d972a..e89fe7147b19 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 instead. tensorflow==2.6.0 -tensorflow-probability==0.14.0 +tensorflow-probability==0.14.1 torch==1.9.0;sys_platform=="darwin" torchvision==0.10.0;sys_platform=="darwin"