mirror of
https://github.com/wassname/ray.git
synced 2026-06-27 21:53:18 +08:00
[Tune] Pin Tune Dependencies (#13027)
Co-authored-by: Ian <ian.rodney@gmail.com>
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
ax-platform==0.1.9; python_version < '3.7'
|
||||
ax-platform==0.1.19; python_version >= '3.7'
|
||||
bayesian-optimization==1.2.0
|
||||
ConfigSpace==0.4.10
|
||||
dragonfly-opt==0.1.6
|
||||
gluoncv==0.9.0
|
||||
gpy==1.9.9
|
||||
gym[atari]==0.18.0
|
||||
h5py==3.1.0
|
||||
hpbandster==0.7.4
|
||||
hyperopt==0.2.5
|
||||
jupyter==1.0.0
|
||||
keras==2.4.3
|
||||
kubernetes==12.0.1
|
||||
lightgbm==3.1.1
|
||||
matplotlib==3.3.3
|
||||
mlflow==1.13.0
|
||||
mxnet==1.7.0.post1
|
||||
nevergrad==0.4.2.post5
|
||||
optuna==2.3.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.0.1
|
||||
tensorflow-probability==0.11.1
|
||||
timm==0.3.2
|
||||
|
||||
|
||||
torch==1.7.0;sys_platform=="darwin"
|
||||
torchvision==0.8.1;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.0+cpu;sys_platform!="darwin"
|
||||
-f https://download.pytorch.org/whl/torch_stable.html
|
||||
torchvision==0.8.1+cpu;sys_platform!="darwin"
|
||||
|
||||
|
||||
transformers==3.1
|
||||
tune-sklearn==0.2.1
|
||||
wandb==0.10.12
|
||||
xgboost==1.3.0.post0
|
||||
xgboost_ray==0.0.1
|
||||
zoopt==0.4.1
|
||||
@@ -1,38 +0,0 @@
|
||||
ax-platform
|
||||
bayesian-optimization
|
||||
ConfigSpace==0.4.10
|
||||
dragonfly-opt
|
||||
gluoncv
|
||||
gorilla # Need this because bug in mlflow. Should be fixed in v1.12.2
|
||||
gym[atari]
|
||||
GPy
|
||||
h5py
|
||||
hpbandster
|
||||
hyperopt>=0.2.5
|
||||
jupyter
|
||||
keras
|
||||
kubernetes
|
||||
lightgbm
|
||||
matplotlib
|
||||
mlflow
|
||||
mxnet
|
||||
nevergrad
|
||||
optuna
|
||||
pytest-remotedata>=0.3.1
|
||||
pytorch-lightning==1.0.3
|
||||
pytorch-lightning-bolts
|
||||
scikit-learn
|
||||
scikit-optimize
|
||||
sigopt
|
||||
smart_open
|
||||
tensorflow-probability
|
||||
timm
|
||||
torch>=1.6.0
|
||||
torchvision>=0.6.0
|
||||
# transformers
|
||||
git+git://github.com/huggingface/transformers.git@bdcc4b78a27775d1ec8f3fd297cb679c257289db#transformers
|
||||
git+git://github.com/ray-project/tune-sklearn@master#tune-sklearn
|
||||
git+git://github.com/ray-project/xgboost_ray@master#xgboost_ray
|
||||
wandb
|
||||
xgboost
|
||||
zoopt>=0.4.1
|
||||
Reference in New Issue
Block a user