google-research

1
# --- Template docker image
2
FROM nvidia/cuda:10.0-cudnn7-runtime
3

4
# --- system packages (not in nvidia image)
5
RUN apt-get update
6
RUN apt-get -y install python3
7
RUN apt-get -y install python3-pip
8

9
# --- (unclear why necessary)
10
WORKDIR /root
11

12
# --- Installs packages
13
RUN pip3 install --upgrade pip
14
RUN pip3 install numpy
15
RUN pip3 install matplotlib
16
RUN pip3 install absl-py
17
RUN pip3 install tqdm
18
RUN pip3 install tensorflow-gpu==2.0.0-alpha0
19

20
# --- Copies the source code
21
RUN mkdir /root/capsule_em
22
COPY ./*.py /root/capsule_em/
23
COPY ./norb/ /root/capsule_em/norb/
24
COPY ./mnist/ /root/capsule_em/mnist/
25

26
# --- Defines the entry point
27
ENTRYPOINT ["python3", "-m", "capsule_em.experiment"]
28