-
Notifications
You must be signed in to change notification settings - Fork 145
/
Dockerfile
23 lines (19 loc) · 656 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
# To build this docker image:
# sudo docker build .
#
# To run the image:
# sudo nvidia-docker run -it <image_id>
# Note: a more recent nvidia/cuda image means users must
# have a more recent CUDA install on their systems
#FROM nvidia/cuda:9.1-cudnn7-devel-ubuntu16.04
FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
# Install dependencies
RUN apt-get update -y
RUN apt-get install -y qt5-default qttools5-dev-tools git python3-pip
RUN pip3 install --upgrade pip
# Clone and install BabyAI git repo
RUN git clone https:/mila-udem/babyai.git
WORKDIR babyai
RUN pip3 install --editable .
# Copy models into the docker image
COPY models models/