summaryrefslogtreecommitdiffhomepage
path: root/benchmarks/workloads/tensorflow/Dockerfile
blob: 262643b98b8a7ca410425dddb6e31e5ae1fc30ee (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
FROM tensorflow/tensorflow:1.13.2

RUN apt-get update \
    && apt-get install -y git
RUN git clone https://github.com/aymericdamien/TensorFlow-Examples.git
RUN python -m pip install -U pip setuptools
RUN python -m pip install matplotlib

WORKDIR /TensorFlow-Examples/examples

ENV PYTHONPATH="$PYTHONPATH:/TensorFlow-Examples/examples"

ENV workload "3_NeuralNetworks/convolutional_network.py"
CMD python ${workload}