diff --git a/test/pipelines-it-local.yml b/test/pipelines-it-local.yml index 31a2e8cdb6..c37ae60382 100644 --- a/test/pipelines-it-local.yml +++ b/test/pipelines-it-local.yml @@ -12,7 +12,7 @@ jobs: python3 -m pip install torch==1.2.0 --user python3 -m pip install torchvision==0.4.0 --user python3 -m pip install keras==2.1.6 --user - python3 -m pip install tensorflow-gpu==1.12.0 --user + python3 -m pip install tensorflow-gpu==1.15 --user sudo apt-get install swig -y PATH=$HOME/.local/bin:$PATH nnictl package install --name=SMAC PATH=$HOME/.local/bin:$PATH nnictl package install --name=BOHB