diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
index 5ec9820522e9..a8bbc60bb3ae 100644
--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
@@ -1,4 +1,5 @@
 # CI docker GPU env
+# tag: v0.41
 FROM nvidia/cuda:8.0-cudnn7-devel
 
 # Base scripts
@@ -48,6 +49,9 @@ RUN bash /install/ubuntu_install_mxnet.sh
 COPY install/ubuntu_install_coreml.sh /install/ubuntu_install_coreml.sh
 RUN bash /install/ubuntu_install_coreml.sh
 
+COPY install/ubuntu_install_tensorflow.sh /install/ubuntu_install_tensorflow.sh
+RUN bash /install/ubuntu_install_tensorflow.sh
+
 COPY install/ubuntu_install_keras.sh /install/ubuntu_install_keras.sh
 RUN bash /install/ubuntu_install_keras.sh
 
diff --git a/docker/install/ubuntu_install_tensorflow.sh b/docker/install/ubuntu_install_tensorflow.sh
index 407954f8fd46..b773fcfb027b 100644
--- a/docker/install/ubuntu_install_tensorflow.sh
+++ b/docker/install/ubuntu_install_tensorflow.sh
@@ -1 +1 @@
-pip3 install tensorflow-gpu
+pip3 install tensorflow