Skip to content

Instantly share code, notes, and snippets.

@prats226
Created February 28, 2018 04:06
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Save prats226/67c23394eda3819eac74d86e55b08a67 to your computer and use it in GitHub Desktop.
Dockerfile for pi training
FROM tensorflow/tensorflow:1.4.1-gpu
RUN apt-get update && apt-get install -y --no-install-recommends \
python-pip \
protobuf-compiler \
python-pil \
python-lxml \
python-numpy \
python-scipy \
python-dev \
python-pip \
python-nose \
g++ \
libopenblas-dev \
git \
python-tk
RUN pip --no-cache-dir install \
lxml \
keras \
theano \
boto3 \
grpcio
WORKDIR /
RUN git clone https://github.com/tensorflow/models.git
WORKDIR /models/research/
RUN protoc object_detection/protos/*.proto --python_out=.
RUN export PYTHONPATH=/tensorflow/models/research/slim:$PYTHONPATH
EXPOSE 6006
ENTRYPOINT /bin/bash
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment