From d0cf8e52e18d60a13e28c46d9333343280181dc2 Mon Sep 17 00:00:00 2001
From: chicm-ms <38930155+chicm-ms@users.noreply.github.com>
Date: Tue, 14 Jan 2020 13:11:32 +0800
Subject: [PATCH] upgrade tensorflow-gpu version for local IT (#1951)

---
 test/pipelines-it-local.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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