forked from alexjc/neural-doodle
-
Notifications
You must be signed in to change notification settings - Fork 2
/
docker-gpu.df
46 lines (41 loc) · 1.55 KB
/
docker-gpu.df
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
FROM nvidia/cuda:7.5-cudnn4-devel
# Install dependencies
RUN apt-get -qq update && \
apt-get -qq install --assume-yes \
"module-init-tools" \
"build-essential" \
"cmake" \
"git" \
"wget" \
"libopenjpeg2" \
"libopenblas-dev" \
"liblapack-dev" \
"libjpeg-dev" \
"libtiff5-dev" \
"zlib1g-dev" \
"libfreetype6-dev" \
"liblcms2-dev" \
"libwebp-dev" \
"gfortran" \
"pkg-config" \
"python3" \
"python3-dev" \
"python3-pip" \
"python3-numpy" \
"python3-scipy" \
"python3-matplotlib" \
"python3-six" \
"python3-networkx" \
"python3-tk" && \
rm -rf /var/lib/apt/lists/* && \
python3 -m pip -q install "cython"
# Install requirements before copying project files
WORKDIR /nd
COPY requirements.txt .
RUN python3 -m pip -q install -r "requirements.txt"
# Copy only required project files
COPY doodle.py .
# Get a pre-trained neural network (VGG19)
RUN wget -q "https://github.com/alexjc/neural-doodle/releases/download/v0.0/vgg19_conv.pkl.bz2"
# Set an entrypoint to the main doodle.py script
ENTRYPOINT ["python3", "doodle.py", "--device=gpu"]