feat: add Dockerfile for local run

2022-06-08 23:22:43 +02:00 · 2022-06-08 23:22:43 +02:00 · 5b16e3aa76
commit 5b16e3aa76
parent fc1945cecd
1 changed files with 15 additions and 0 deletions
--- a/15
+++ b/15
@ -0,0 +1,15 @@
+FROM docker.io/tensorflow/tensorflow:2.6.0
+
+COPY requirements.txt .
+RUN pip3 install --upgrade pip==20.0.2 && pip3 list && pip3 install -r requirements.txt \
+    && 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
+
+