From 4025496783943490a67b850725d7a394c0205843 Mon Sep 17 00:00:00 2001
From: galleibo-intel <gal.leibovich@intel.com>
Date: Mon, 5 Feb 2018 15:47:06 +0200
Subject: [PATCH] Setting tensorflow-gpu version to 1.4.1 (1.5.0 is not tested
 yet)

---
 install.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/install.sh b/install.sh
index b3e3286..432f949 100755
--- a/install.sh
+++ b/install.sh
@@ -200,5 +200,5 @@ if ! [ -x "$(command -v nvidia-smi)" ]; then
     pip3 install https://anaconda.org/intel/tensorflow/1.3.0/download/tensorflow-1.3.0-cp35-cp35m-linux_x86_64.whl
 else
     # GPU supported TensorFlow
-    pip3 install tensorflow-gpu
+    pip3 install tensorflow-gpu==1.4.1
 fi