volcano.sh/volcano@v1.9.0/example/integrations/tensorflow/benchmark/Dockerfile (about)

     1  # NOTE: the build process would change during developing,
     2  # the commit ID when first creating the image: 62c833f806db621943a6cf8195657b9d0fa67d93 (master)
     3  # original image is: gcr.io/kubeflow/tf-benchmarks-cpu:v20171202-bdab599-dirty-284af3,
     4  # the image needs an update to use the latest tf-benchmark logic
     5  # ref => https://github.com/tensorflow/benchmarks/tree/master/scripts/tf_cnn_benchmarks.
     6  FROM python:2.7
     7  MAINTAINER volcano <volcano-sh@googlegroups.com>
     8  RUN  apt-get update --fix-missing \
     9  && apt-get install -y git \
    10  && apt-get clean \
    11  && rm -rf /var/lib/apt/lists/*
    12  RUN pip install tf-nightly-gpu \
    13  && git clone https://github.com/tensorflow/benchmarks.git /opt/tf-benchmarks