robocar-training/Dockerfile.gpu

FROM python:3.5 as builder

RUN mkdir -p /usr/src
ADD . /usr/src
WORKDIR /usr/src

RUN python3 setup.py sdist

#FROM tensorflow/tensorflow:1.8.0-py3
FROM tensorflow/tensorflow:1.15.0-gpu-py3

#tensorflow-serving-api-python3==1.7.0
COPY requirements.txt .
RUN pip3 install --upgrade pip==20.0.2 && pip3 list && pip3 install -r requirements.txt \
    && pip3 list

WORKDIR /root

RUN apt-get -y update && \
    apt-get -y install curl && \
    apt-get -y install vim && \
    apt-get -y install iputils-ping && \
    apt-get -y install nginx

# install telegraf
RUN cd /tmp && \
    curl -O https://dl.influxdata.com/telegraf/releases/telegraf_1.4.2-1_amd64.deb && \
    dpkg -i telegraf_1.4.2-1_amd64.deb && \
    cd -

COPY --from=builder /usr/src/dist/robocars_sagemaker_container-1.0.0.tar.gz .

RUN pip3 install robocars_sagemaker_container-1.0.0.tar.gz

RUN rm robocars_sagemaker_container-1.0.0.tar.gz

ENTRYPOINT ["train"]