15 lines
418 B
Docker
15 lines
418 B
Docker
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}
|