mirror of https://github.com/kubeflow/examples.git
58 lines
1.9 KiB
Docker
58 lines
1.9 KiB
Docker
# Copyright 2018 Google Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# https://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.#
|
|
|
|
ARG base_image=tensorflow/tensorflow:1.8.0-gpu
|
|
|
|
FROM $base_image
|
|
|
|
MAINTAINER Yixin Shi <yxshi@google.com>
|
|
|
|
RUN apt-get update && apt-get install -y software-properties-common && \
|
|
add-apt-repository ppa:ubuntu-toolchain-r/test -y && \
|
|
apt-get update && apt-get install -y \
|
|
build-essential \
|
|
curl \
|
|
libcurl3-dev \
|
|
git \
|
|
libfreetype6-dev \
|
|
libpng12-dev \
|
|
libzmq3-dev \
|
|
pkg-config \
|
|
python-dev \
|
|
python-numpy \
|
|
python-pip \
|
|
swig \
|
|
zip \
|
|
zlib1g-dev \
|
|
&& \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
WORKDIR /root
|
|
|
|
RUN pip install -U setuptools && \
|
|
curl -o /tmp/kubeflow-batch-predict.zip -L https://github.com/kubeflow/batch-predict/archive/master.zip && \
|
|
mkdir -p /tmp/unzipped && \
|
|
unzip -o /tmp/kubeflow-batch-predict.zip -d /tmp/unzipped && \
|
|
sed -i 's/tensorflow/tensorflow-gpu/g' /tmp/unzipped/batch-predict-master/kubeflow_batch_predict/version.py && \
|
|
cd /tmp/unzipped && \
|
|
zip -r /opt/kubeflow-batch-predict.zip * && \
|
|
pip install /opt/kubeflow-batch-predict.zip && \
|
|
pip install apache-beam[gcp]==2.3.0 && \
|
|
rm -rf /tmp/unzipped && \
|
|
rm -rf /root/.cache/pip;
|
|
|
|
ENTRYPOINT [ "python", "-m", "kubeflow_batch_predict.dataflow.batch_prediction_main" ]
|