ray/python/requirements/ml/requirements_tune.txt
Amog Kamsetty 84e958f330
[ML] Consolidate and upgrade Deep Learning Dependencies (#18574)
* wip
'

* upgrade requirements

* add file

* fix

* fixes

* Apply suggestions from code review

Try mlagents==0.21.0 for now (works with torch 1.9).

* Apply suggestions from code review

* wip

* wip

* fix

* fix

* upgrade lightning bolts

* address comment

Co-authored-by: Sven Mika <sven@anyscale.io>
2021-09-16 20:16:40 -07:00

47 lines
1.1 KiB
Text

-r requirements_dl.txt
autogluon.core==0.1.0
ax-platform==0.1.9; python_version < '3.7'
ax-platform==0.2.1; python_version >= '3.7'
bayesian-optimization==1.2.0
ConfigSpace==0.4.18
dask[complete]==2021.03.0; python_version < '3.7'
dask[complete]==2021.8.1; python_version >= '3.7'
dragonfly-opt==0.1.6
flaml==0.5.2
freezegun==1.1.0
gluoncv==0.10.1.post0
gpy==1.10.0
gym[atari]==0.18.3
h5py==3.1.0
hpbandster==0.7.4
pymoo<0.5.0 # this is a HEBO dependency, remove after https://github.com/huawei-noah/noah-research/issues/41 is fixed
HEBO==0.1.0
hyperopt==0.2.5
jupyter==1.0.0
kubernetes==17.17.0
lightgbm==3.2.1
matplotlib==3.3.4; python_version < '3.7'
matplotlib==3.4.3; python_version >= '3.7'
mlflow==1.19.0
mxnet==1.8.0.post0
nevergrad==0.4.3.post7
optuna==2.9.1
pytest-remotedata==0.3.2
lightning-bolts==0.4.0
pytorch-lightning==1.4.5
scikit-learn==0.24.2
scikit-optimize==0.8.1
sigopt==7.5.0
smart_open==5.1.0
timm==0.4.5
transformers==4.9.1
wandb==0.10.29
xgboost==1.3.3
zoopt==0.4.1
# Dependencies for Hugging Face examples:
# `python/ray/util/sgd/v2/examples/transformers/transformers_example.py`
accelerate==0.3.0
datasets==1.11.0
sentencepiece==0.1.96