robocar-training/Dockerfile.gpu

38 lines
935 B
Docker

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.8.0-gpu-py3
#tensorflow-serving-api-python3==1.7.0
RUN pip3 list && pip3 install numpy boto3 six awscli flask==0.11 Jinja2==2.9 gevent gunicorn keras==2.1.3 pillow h5py \
&& 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"]