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