ray/python/requirements/requirements_tune.txt
Kai Fricke 7364a7a327
[tune] Move Optuna to ask(fixed_distributions) interface (#14731)
Adjusting to changes in Optuna 2.6.0. Old interface was marked as deprecated.
2021-03-22 12:25:37 +01:00

47 lines
1.1 KiB
Text

autogluon.core==0.0.16b20210113 # to pin autogluon.core
ax-platform==0.1.9; python_version < '3.7'
ax-platform==0.1.20; python_version >= '3.7'
bayesian-optimization==1.2.0
ConfigSpace==0.4.10
dragonfly-opt==0.1.6
gluoncv==0.9.1
gpy==1.9.9
gym[atari]==0.18.0
h5py==3.1.0
hpbandster==0.7.4
HEBO==0.1.0
hyperopt==0.2.5
jupyter==1.0.0
keras==2.4.3
kubernetes==12.0.1
lightgbm==3.1.1
matplotlib==3.3.4
mlflow==1.14.0
mxnet==1.7.0.post1
nevergrad==0.4.2.post5
optuna==2.6.0
pytest-remotedata==0.3.2
pytorch-lightning-bolts==0.2.5
pytorch-lightning==1.0.3
scikit-learn==0.22.2
scikit-optimize==0.8.1
sigopt==5.7.0
smart_open==4.2.0
tensorflow-probability==0.11.1
timm==0.3.2
torch==1.7.1;sys_platform=="darwin"
torchvision==0.8.2;sys_platform=="darwin"
# On non-OSX machines only install CPU version of torch and torchvision
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.7.1+cpu;sys_platform!="darwin"
-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.8.2+cpu;sys_platform!="darwin"
transformers==4.3.2
wandb==0.10.19
xgboost==1.3.3
zoopt==0.4.1