blob: b5763e8ae29602d09da989fb8053b10b63a9fc9d (
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 --depth 1 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}
|
