[DOCKER] add ci tag, upgrade gpu-ci (apache#2438)

neo-ai · Feb 20, 2019 · 0c3d887 · 0c3d887
1 parent ea6e00a
commit 0c3d887
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git 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
@@ -1 +1 @@
-pip3 install tensorflow-gpu
+pip3 install tensorflow
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		pip3 install tensorflow-gpu
		pip3 install tensorflow