diff --git a/python/requirements/requirements_tune.txt b/python/requirements/requirements_tune.txt new file mode 100644 index 000000000000..86f8d9e7a700 --- /dev/null +++ b/python/requirements/requirements_tune.txt @@ -0,0 +1,45 @@ +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.1 +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.1 +mxnet==1.7.0.post1 +nevergrad==0.4.2.post5 +optuna==2.4.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 +wandb==0.10.18 +xgboost==1.3.0.post0 +zoopt==0.4.1