upgrade tensorflow-gpu version for local IT (#1951)

d0cf8e52 · chicm-ms · QuanluZhang · 30152b04 · d0cf8e52
Commit d0cf8e52 authored Jan 14, 2020 by chicm-ms Committed by QuanluZhang Jan 14, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/pipelines-it-local.yml test/pipelines-it-local.yml +1 -1

No files found.
--- 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