-
Notifications
You must be signed in to change notification settings - Fork 2
/
Dockerfile
59 lines (49 loc) · 1.4 KB
/
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
FROM ubuntu:16.04
ENV DEBIAN_FRONTEND noninteractive
# Ubuntu packages + Numpy
RUN apt-get update \
&& apt-get install -y --no-install-recommends \
apt-utils \
build-essential \
g++ \
git \
curl \
cmake \
zlib1g-dev \
libjpeg-dev \
xvfb \
libav-tools \
xorg-dev \
libboost-all-dev \
libsdl2-dev \
swig \
python3 \
python3-dev \
python3-future \
python3-pip \
python3-setuptools \
python3-wheel \
python3-tk \
libopenblas-base \
libatlas-dev \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*
# Upgrade pip
RUN python3 -m pip install --upgrade pip
RUN pip install --ignore-installed pip
# Install Pytorch
RUN git clone https://github.com/pytorch/vision.git && cd vision && pip install -v .
# Install Python packages - Step 1
COPY requirements.txt /tmp/
RUN python3 -m pip install -r /tmp/requirements.txt
# Install OPEN AI GYM environments
RUN git clone https://github.com/openai/gym.git \
&& cd gym \
&& pip install -e '.[atari]' \
&& pip install -e '.[classic_control]' \
&& pip install -e '.[box2d]'
# Command for jupyter themes (comment if you dont use jupyter themes)
RUN jt -t onedork -fs 11 -altp -tfs 12 -nfs 12 -ofs 10 -cellw 95% -T -cursc r
# Add directory
RUN mkdir /ds
ENV DEBIAN_FRONTEND teletype