forked from javawizard/arts
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDockerfile
More file actions
64 lines (54 loc) · 1.6 KB
/
Dockerfile
File metadata and controls
64 lines (54 loc) · 1.6 KB
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
60
61
62
63
FROM ubuntu:14.04
MAINTAINER Duane Johnson <duane.johnson@gmail.com>
ADD downloads/cuda_7.5.18_linux.run /tmp/cuda_install/cuda.run
ADD downloads/NVIDIA-Linux-x86_64-352.63.run /tmp/cuda_install/driver.run
ENV INSTALL_PREFIX /usr/local
ENV CUDA_PREFIX /usr/local/cuda-7.5
ENV CUDNN_ENABLED 1
# Install deb packages
RUN apt-get update && apt-get install -y --force-yes \
build-essential \
git \
libopenblas-dev \
python-dev \
python-numpy \
python-scipy \
python-setuptools \
vim \
wget
# Install cuda base library
RUN cd /tmp/cuda_install && \
# Make the run file executable and extract
chmod +x cuda.run && sync && \
chmod +x driver.run && sync && \
./cuda.run -extract=`pwd` && \
# Install CUDA drivers (silent, no kernel)
./driver.run -s --no-kernel-module && \
# Install toolkit (silent)
./cuda-linux64-rel-*.run -noprompt
# Add to path
ENV PATH=/usr/local/cuda-7.5/bin:$PATH \
LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
# Install cuDNN
ADD downloads/cuda /usr/local/cuda-7.5
RUN easy_install cython
# Install cudarray
ADD cudarray /tmp/cudarray
RUN cd /tmp/cudarray && \
make && make install && \
python setup.py install
# Install Pillow image library dependencies
# RUN apt-get update && apt-get install -y --force-yes \
# libjpeg-dev \
# libfreetype6-dev \
# python-liblcms
RUN apt-get install -y --force-yes \
libjpeg-dev \
libfreetype6-dev \
python-liblcms
# Install deeppy
ADD deeppy /tmp/deeppy
RUN cd /tmp/deeppy && \
python setup.py install
ADD neural_artistic_style /style
ADD downloads/imagenet-vgg-verydeep-19.mat /style/imagenet-vgg-verydeep-19.mat