2022-06-08 21:20:39 +00:00
|
|
|
FROM docker.io/tensorflow/tensorflow:2.6.0-gpu
|
2018-08-03 10:23:18 +00:00
|
|
|
|
2020-03-02 18:21:21 +00:00
|
|
|
COPY requirements.txt .
|
|
|
|
|
RUN pip3 install --upgrade pip==20.0.2 && pip3 list && pip3 install -r requirements.txt \
|
2018-08-03 10:30:42 +00:00
|
|
|
&& pip3 list
|
2018-08-03 10:23:18 +00:00
|
|
|
|
|
|
|
|
WORKDIR /root
|
|
|
|
|
|
2022-06-08 21:20:39 +00:00
|
|
|
# copy the training script inside the container
|
|
|
|
|
COPY src/tf_container/train.py /opt/ml/code/train.py
|
2018-08-03 10:23:18 +00:00
|
|
|
|
2022-06-08 21:20:39 +00:00
|
|
|
# define train.py as the script entry point
|
|
|
|
|
ENV SAGEMAKER_PROGRAM train.py
|
2018-08-03 10:23:18 +00:00
|
|
|
|
|
|
|
|
|
