-
Notifications
You must be signed in to change notification settings - Fork 2
/
Dockerfile
25 lines (18 loc) · 893 Bytes
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
FROM duckietown/challenge-aido_lf-template-tensorflow:daffy
# let's create our workspace, we don't want to clutter the container
RUN rm -r /workspace; mkdir /workspace
WORKDIR /workspace
# let's copy all our solution files to our workspace
# if you have more file use the COPY command to move them to the workspace
# IMPORTANT: This should fail if "frozen_graph.pb" does not exist.
# This file will contain a learned model which the agent should execute.
COPY learned_models/frozen_graph.pb .
COPY requirements.txt .
COPY src/ src/
# here, we install the requirements, some requirements come by default
# you can add more if you need to in requirements.txt
RUN pip install --no-cache-dir -r requirements.txt
# necessary to get opencv to work properly
RUN apt update && apt install -y libsm6 libxext6
# CMD python solution.py
ENTRYPOINT ["python", "-m", "src.imitation.imitation_agent"]