robocar-training/Dockerfile.gpu

39 lines
930 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.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"]