aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/dist_test/Dockerfile
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/tools/dist_test/Dockerfile')
-rw-r--r--tensorflow/tools/dist_test/Dockerfile28
1 files changed, 28 insertions, 0 deletions
diff --git a/tensorflow/tools/dist_test/Dockerfile b/tensorflow/tools/dist_test/Dockerfile
new file mode 100644
index 0000000000..fba23af55d
--- /dev/null
+++ b/tensorflow/tools/dist_test/Dockerfile
@@ -0,0 +1,28 @@
+FROM ubuntu:14.04
+
+MAINTAINER Shanqing Cai <cais@google.com>
+
+RUN apt-get update
+RUN apt-get install -y \
+ bc \
+ curl \
+ python \
+ python-numpy \
+ python-pip
+
+# Install Google Cloud SDK
+RUN curl -O https://dl.google.com/dl/cloudsdk/channels/rapid/install_google_cloud_sdk.bash
+RUN chmod +x install_google_cloud_sdk.bash
+RUN ./install_google_cloud_sdk.bash --disable-prompts --install-dir=/var/gcloud
+
+# Install kubectl
+RUN /var/gcloud/google-cloud-sdk/bin/gcloud components install kubectl
+
+# Install nightly TensorFlow pip
+# TODO(cais): Should we build it locally instead?
+RUN pip install \
+ http://ci.tensorflow.org/view/Nightly/job/nightly-matrix-cpu/TF_BUILD_CONTAINER_TYPE=CPU,TF_BUILD_IS_OPT=OPT,TF_BUILD_IS_PIP=PIP,TF_BUILD_PYTHON_VERSION=PYTHON2,label=cpu-slave/lastSuccessfulBuild/artifact/pip_test/whl/tensorflow-0.7.1-cp27-none-linux_x86_64.whl
+
+# Copy test files
+COPY scripts /var/tf-dist-test/scripts
+COPY python /var/tf-dist-test/python