-
Notifications
You must be signed in to change notification settings - Fork 20
/
Copy pathDockerfile
62 lines (53 loc) · 1.38 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
60
61
62
FROM debian
RUN apt-get clean && apt-get update && apt-get install -y \
bc \
ca-certificates \
curl \
cython \
g++ \
git \
ipython \
libatlas-base-dev \
libatlas-dev \
libboost-all-dev \
libgflags-dev \
libgoogle-glog-dev \
libhdf5-dev \
libleveldb-dev \
liblmdb-dev \
libopencv-dev \
libprotobuf-dev \
libsnappy-dev \
make \
protobuf-compiler \
python-dateutil \
python-gflags \
python-h5py \
python-leveldb \
python-matplotlib \
python-networkx \
python-nose \
python-numpy \
python-pandas \
python-pil \
python-protobuf \
python-scipy \
python-skimage-lib \
python-yaml \
--no-install-recommends \
&& rm -rf /var/lib/apt/lists/*
RUN curl https://bootstrap.pypa.io/get-pip.py | python
RUN pip install scikit-image
RUN cd root && git clone --depth 1 --single-branch https://github.com/BVLC/caffe.git
RUN curl http://dl.caffe.berkeleyvision.org/bvlc_googlenet.caffemodel > /root/caffe/models/bvlc_googlenet/bvlc_googlenet.caffemodel
RUN cd /root/caffe && \
cp Makefile.config.example Makefile.config && \
sed -i 's/# CPU_ONLY/CPU_ONLY/g' Makefile.config && \
echo 'INCLUDE_DIRS += /usr/include/hdf5/serial' >> Makefile.config && \
echo 'LIBRARY_DIRS += /usr/lib/x86_64-linux-gnu/hdf5/serial' >> Makefile.config && \
make -j"$(nproc)" all pycaffe
ENV PYTHONPATH=/root/caffe/python
WORKDIR /ddd
COPY start.sh /ddd/
COPY deepdreams.py /ddd/
ENTRYPOINT ["/ddd/start.sh"]