diff --git a/python/requirements/requirements_tune.txt b/python/requirements/requirements_tune.txt index 0628c6966a37..1db359f3c4ce 100644 --- a/python/requirements/requirements_tune.txt +++ b/python/requirements/requirements_tune.txt @@ -22,7 +22,7 @@ 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 +pytorch-lightning==1.2.8 scikit-learn==0.22.2 scikit-optimize==0.8.1 sigopt==5.7.0 @@ -31,14 +31,14 @@ tensorflow-probability==0.11.1 timm==0.3.2 -torch==1.7.1;sys_platform=="darwin" -torchvision==0.8.2;sys_platform=="darwin" +torch==1.8.1;sys_platform=="darwin" +torchvision==0.9.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.1+cpu;sys_platform!="darwin" +torch==1.8.1+cpu;sys_platform!="darwin" -f https://download.pytorch.org/whl/torch_stable.html -torchvision==0.8.2+cpu;sys_platform!="darwin" +torchvision==0.9.1+cpu;sys_platform!="darwin" transformers==4.3.2