9 lines
294 B
Docker
9 lines
294 B
Docker
ARG BASE_IMAGE_TAG=1.12.0-py3
|
|
FROM tensorflow/tensorflow:$BASE_IMAGE_TAG
|
|
COPY requirements.txt .
|
|
RUN python3 -m pip install -r \
|
|
requirements.txt --quiet --no-cache-dir \
|
|
&& rm -f requirements.txt
|
|
COPY ./src /pipelines/component/src
|
|
ENTRYPOINT python3 /pipelines/component/src/train.py
|