Docker py-faster-rcnn and caffe with cuda8

gistfile1.txt

FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04

ENV PACKAGES_ROOT=/opt

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 \
    python-protobuf \
    libsnappy-dev \
    protobuf-compiler \
    python-dev \
    python-numpy \
    python-pip \
    python-setuptools \
    python-scipy \
    gfortran \
    python-opencv \
    cython \
    python-skimage \
    python-yaml && \
    rm -rf /var/lib/apt/lists/*

RUN pip install --upgrade pip==9.0
RUN pip install easydict matplotlib

WORKDIR $PACKAGES_ROOT

RUN git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git && \
    cd $PACKAGES_ROOT/py-faster-rcnn && \
    git checkout 96dc9f1d

WORKDIR $PACKAGES_ROOT/py-faster-rcnn/lib

RUN sed -i 's/sm_35/sm_30/g' setup.py && \
    rm -f utils/bbox.c nms/cpu_nms.c nms/gpu_nms.cpp && \
    make

WORKDIR $PACKAGES_ROOT/py-faster-rcnn/caffe-fast-rcnn

RUN git remote add acmiyaguchi https://github.com/acmiyaguchi/caffe-fast-rcnn.git
RUN git fetch acmiyaguchi
RUN git checkout acmiyaguchi/faster-rcnn-rebased

RUN cp Makefile.config.example Makefile.config && \
    echo "USE_CUDNN := 1" >> Makefile.config && \
    echo 'INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/local/cuda/include /usr/include/hdf5/serial/' >> Makefile.config && \
    echo 'LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/' >> Makefile.config && \
    echo 'WITH_PYTHON_LAYER := 1' >> Makefile.config && \
    echo 'LIBRARIES += hdf5_serial_hl hdf5_serial' >> Makefile.config && \
    sed -i '/compute_35/d' Makefile.config && \
    cat Makefile.config  && \
    make -j8  && make pycaffe -j8

RUN sed -i 's/self.param_str_/self.param_str/g' /opt/py-faster-rcnn/lib/rpn/proposal_layer.py

WORKDIR $PACKAGES_ROOT/py-faster-rcnn