-
Notifications
You must be signed in to change notification settings - Fork 4
/
Dockerfile
34 lines (31 loc) · 996 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
24
25
26
27
28
29
30
31
32
33
34
FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04
LABEL maintainer [email protected]
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
cmake \
git \
wget \
libatlas-base-dev \
libboost-all-dev \
libgflags-dev \
libgoogle-glog-dev \
libhdf5-serial-dev \
libleveldb-dev \
liblmdb-dev \
libopencv-dev \
libprotobuf-dev \
libsnappy-dev \
protobuf-compiler \
python-dev \
python-numpy \
python-pip \
python-opencv \
python-setuptools \
python-scipy && \
rm -rf /var/lib/apt/lists/*
RUN pip install easydict
RUN pip install cython
RUN git clone --recursive https://github.com/phtruongan/py-faster-rcnn.git
RUN cd py-faster-rcnn-docker/lib && make
RUN cd py-faster-rcnn-docker/caffe-fast-rcnn && make && make pycaffe
RUN cd py-faster-rcnn-docker && ./data/scripts/fetch_faster_rcnn_models.sh