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

microsoft · Jan 14, 2020 · d0cf8e5 · d0cf8e5
1 parent 30152b0
commit d0cf8e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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