robocars/robocar-training
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: add Dockerfile for local run

Browse Source
This commit is contained in:
Cyrille Nofficial 2022-06-08 23:22:43 +02:00
parent fc1945cecd
commit 5b16e3aa76
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
Dockerfile Normal file
View File

@ -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