From adb8d77b2bc7f1baeb2d5f7e85b98e8a07d66eb2 Mon Sep 17 00:00:00 2001 From: Amog Kamsetty Date: Fri, 5 Nov 2021 00:58:34 -0700 Subject: [PATCH] [Deps] Bump tensorflow on Docker image and add Codeowners (#20041) --- .github/CODEOWNERS | 4 ++++ python/requirements/ml/requirements_dl.txt | 4 ++++ python/requirements_ml_docker.txt | 9 +++++++-- 3 files changed, 15 insertions(+), 2 deletions(-) diff --git a/.github/CODEOWNERS b/.github/CODEOWNERS index c9d4191dd..3b7090591 100644 --- a/.github/CODEOWNERS +++ b/.github/CODEOWNERS @@ -52,6 +52,10 @@ # RLlib. #/python/ray/rllib/ @ray-project/rllib +# ML Docker Dependencies +/python/requirements/ml/requirements_dl.txt @amogkam @sven1977 @richardliaw @matthewdeng +/python/requirements_ml_docker.txt @amogkam @sven1977 @richardliaw @matthewdeng + # ==== Build and CI ==== # Bazel. diff --git a/python/requirements/ml/requirements_dl.txt b/python/requirements/ml/requirements_dl.txt index b7c81334f..3970fa10a 100644 --- a/python/requirements/ml/requirements_dl.txt +++ b/python/requirements/ml/requirements_dl.txt @@ -1,6 +1,10 @@ # 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. + +# If you make changes to this file, please also change `requirements_ml_docker.txt`! + + # TODO(amogkam): Remove after https://github.com/tensorflow/tensorflow/issues/52922 is fixed. keras==2.6.0 tensorflow==2.6.0 diff --git a/python/requirements_ml_docker.txt b/python/requirements_ml_docker.txt index 70c4413e8..166d2bab9 100644 --- a/python/requirements_ml_docker.txt +++ b/python/requirements_ml_docker.txt @@ -3,9 +3,14 @@ ipython # Needed for Ray Client error message serialization/deserialization. tblib +# If you make changes to anything below this line, please also change `requirements_dl.txt`! + +# TODO(amogkam): Remove after https://github.com/tensorflow/tensorflow/issues/52922 is fixed. +keras==2.6.0 # In TF >v2, GPU support is included in the base package. -tensorflow==2.5.0 -tensorflow-probability==0.13.0 +tensorflow==2.6.0 +tensorflow-probability==0.14.0 + -f https://download.pytorch.org/whl/torch_stable.html torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html