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

9e9cda54 · Tianqi Chen · GitHub · f7beea4b · 9e9cda54 · 9e9cda54
Unverified Commit 9e9cda54 authored Jan 15, 2019 by Tianqi Chen Committed by GitHub Jan 15, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletions

docker/Dockerfile.ci_gpu
+4 -0

docker/install/ubuntu_install_tensorflow.sh
+1 -1

No files found.
--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
 # 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


--- a/docker/install/ubuntu_install_tensorflow.sh
+++ b/docker/install/ubuntu_install_tensorflow.sh
-pip3 install tensorflow-gpu
+pip3 install tensorflow