2022-07-31 12:01:48 +00:00
|
|
|
FROM docker.io/tensorflow/tensorflow:2.9.0
|
2022-06-08 21:22:43 +00:00
|
|
|
|
|
|
|
|
COPY requirements.txt .
|
2022-07-31 12:01:48 +00:00
|
|
|
RUN pip3 install --upgrade pip==22.2.1 && pip3 list && pip3 install -r requirements.txt \
|
2022-06-08 21:22:43 +00:00
|
|
|
&& pip3 list
|
|
|
|
|
|
|
|
|
|
WORKDIR /root
|
|
|
|
|
|
|
|
|
|
# copy the training script inside the container
|
|
|
|
|
COPY src/tf_container/train.py /opt/ml/code/train.py
|
|
|
|
|
|
|
|
|
|
# define train.py as the script entry point
|
|
|
|
|
ENV SAGEMAKER_PROGRAM train.py
|
|
|
|
|
|
|
|
|
|
|
