From 9e9cda54cbf339c0874c34dd0c66d0fdd8cdb371 Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Tue, 15 Jan 2019 14:29:13 -0800 Subject: [PATCH] [DOCKER] add ci tag, upgrade gpu-ci (#2438) --- docker/Dockerfile.ci_gpu | 4 ++++ docker/install/ubuntu_install_tensorflow.sh | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index 5ec982052..a8bbc60bb 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 407954f8f..b773fcfb0 100644 --- a/docker/install/ubuntu_install_tensorflow.sh +++ b/docker/install/ubuntu_install_tensorflow.sh @@ -1 +1 @@ -pip3 install tensorflow-gpu +pip3 install tensorflow -- 2.34.1