docker-file gpu

Dockerfile

FROM gcr.io/tensorflow/tensorflow:latest-gpu-py3

ENV HOME_PATH=/home/ubuntu \
    APP_PATH=/home/ubuntu/recognizer

RUN DEBIAN_FRONTEND=noninteractive apt-get update && apt-get install -y \
    build-essential git
RUN pip install jwcrypto==0.4.2
RUN pip install cython

WORKDIR $APP_PATH
RUN mkdir -p $APP_PATH
COPY ./ $APP_PATH
RUN pip install -r requirement.txt


COPY ./dev/.ssh /root/.ssh
RUN chmod 600 /root/.ssh/config \
    && chmod 600 /root/.ssh/id_rsa

WORKDIR $APP_PATH/darkflow

RUN python setup.py build_ext --inplace
RUN pip install .

WORKDIR $APP_PATH