[docker] Refactor "autoscaler" image into "-autoscaler" tag and "ray-ml" image. (#10351)

2026-07-02 07:55:45 +08:00 · 2020-09-02 13:03:35 -07:00
parent 57c4183724
commit 4324dd5929
11 changed files with 78 additions and 54 deletions
@@ -1,20 +0,0 @@
-bayesian-optimization
-ConfigSpace==0.4.10
-google-api-python-client
-google-oauth
-h5py
-hpbandster
-hyperopt==0.1.2
-ipython
-keras
-lightgbm
-mlflow
-nevergrad
-oauth2client
-scikit-optimize
-sigopt
-smart_open
-tensorflow_probability
-torch
-torchvision
-xgboost
@@ -0,0 +1,3 @@
+-r requirements_rllib.txt
+-r requirements_tune.txt
+ipython
@@ -1,5 +1,7 @@
 tensorflow-probability
 gast
-torch
-torchvision
+# Version requirement to match Tune
+torch>=1.5.0
+# Version requirement to match Tune
+torchvision>=0.6.0
 smart_open
@@ -21,7 +21,7 @@ pytorch-lightning
 scikit-optimize
 sigopt
 smart_open
-tensorflow_probability
+tensorflow-probability
 timm
 torch>=1.5.0
 torchvision>=0.6.0