diff --git a/python/requirements/ml/requirements_dl.txt b/python/requirements/ml/requirements_dl.txt index 1f97dc66cadb..b4c89ca2522f 100644 --- a/python/requirements/ml/requirements_dl.txt +++ b/python/requirements/ml/requirements_dl.txt @@ -1,7 +1,7 @@ # These requirements are used for the CI and CPU-only Docker images so we install CPU only versions of torch. # For GPU Docker images, you should install requirements_ml_docker.txt instead. -tensorflow==2.5.0 +tensorflow==2.6.0 tensorflow-probability==0.13.0 torch==1.9.0;sys_platform=="darwin"