diff --git a/test/pipelines/pipelines-it-local.yml b/test/pipelines/pipelines-it-local.yml index 00fb8e11e5..c66a9b7186 100644 --- a/test/pipelines/pipelines-it-local.yml +++ b/test/pipelines/pipelines-it-local.yml @@ -14,7 +14,7 @@ jobs: python3 -m pip install torchvision==0.4.1 --user python3 -m pip install torch==1.3.1 --user python3 -m pip install keras==2.1.6 --user - python3 -m pip install tensorflow-gpu==1.15 --user + python3 -m pip install tensorflow-gpu==1.15.2 --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