hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-05 10:01:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Deps] Bump tensorflow on Docker image and add Codeowners (#20041)

Browse source
This commit is contained in:
Amog Kamsetty 2021-11-05 00:58:34 -07:00 • committed by GitHub
parent 60e9737679
commit adb8d77b2b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 15 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
.github/CODEOWNERS vendored
View file

@ -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.

4
python/requirements/ml/requirements_dl.txt
View file

@ -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

9
python/requirements_ml_docker.txt
View file

@ -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