# Don't edit this file directly, since it was generated from a template,

# and you're changes will be *clobbered*. Edit the template instead.

{{ if .GPU }}

FROM tleyden5iwx/ubuntu-cuda

{{ else }}

FROM ubuntu:14.04

{{ end }}

ENV PYTHONPATH /opt/caffe/python

# Add caffe binaries to path

ENV PATH $PATH:/opt/caffe/.build_release/tools

# Get dependencies

RUN apt-get update && apt-get install -y \

bc \

cmake \

curl \

gcc-4.6 \

g++-4.6 \

gcc-4.6-multilib \

g++-4.6-multilib \

gfortran \

git \

libprotobuf-dev \

libleveldb-dev \

libsnappy-dev \

libopencv-dev \

libboost-all-dev \

libhdf5-serial-dev \

liblmdb-dev \

libjpeg62 \

libfreeimage-dev \

libatlas-base-dev \

pkgconf \

protobuf-compiler \

python-dev \

python-pip \

unzip \

wget

# Use gcc 4.6

RUN update-alternatives --install /usr/bin/cc cc /usr/bin/gcc-4.6 30 && \

update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++-4.6 30 && \

update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.6 30 && \

update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.6 30

{{ if .GPU }}

# Allow it to find CUDA libs

RUN echo "/usr/local/cuda/lib64" > /etc/ld.so.conf.d/cuda.conf && \

ldconfig

{{ end }}

# Clone the Caffe repo

RUN cd /opt && git clone https://github.com/BVLC/caffe.git

{{ if .DEVELOP_BRANCH }}

RUN cd /opt/caffe && git checkout -t origin/dev

{{ end }}

# Glog

RUN cd /opt && wget https://google-glog.googlecode.com/files/glog-0.3.3.tar.gz && \

tar zxvf glog-0.3.3.tar.gz && \

cd /opt/glog-0.3.3 && \

./configure && \

make && \

make install

# Workaround for error loading libglog:

# error while loading shared libraries: libglog.so.0: cannot open shared object file

# The system already has /usr/local/lib listed in /etc/ld.so.conf.d/libc.conf, so

# running `ldconfig` fixes the problem (which is simpler than using $LD_LIBRARY_PATH)

# TODO: looks like this needs to be run _every_ time a new docker instance is run,

# so maybe LD_LIBRARY_PATh is a better approach (or add call to ldconfig in ~/.bashrc)

RUN ldconfig

# Gflags

RUN cd /opt && \

wget https://github.com/schuhschuh/gflags/archive/master.zip && \

unzip master.zip && \

cd /opt/gflags-master && \

mkdir build && \

cd /opt/gflags-master/build && \

export CXXFLAGS="-fPIC" && \

cmake .. && \

make VERBOSE=1 && \

make && \

make install

# Build Caffe core

RUN cd /opt/caffe && \

cp Makefile.config.example Makefile.config && \

{{ if .CPU }} echo "CPU_ONLY := 1" >> Makefile.config && \ {{ else }} \ {{ end }}

echo "CXX := /usr/bin/g++-4.6" >> Makefile.config && \

sed -i 's/CXX :=/CXX ?=/' Makefile && \

make all

# Add ld-so.conf so it can find libcaffe.so

ADD caffe-ld-so.conf /etc/ld.so.conf.d/

# Run ldconfig again (not sure if needed)

RUN ldconfig

# Install python deps

RUN cd /opt/caffe && \

cat python/requirements.txt | xargs -L 1 sudo pip install

# Numpy include path hack - github.com/BVLC/caffe/wiki/Ubuntu-14.04-VirtualBox-VM

RUN ln -s /usr/include/python2.7/ /usr/local/include/python2.7 && \

ln -s /usr/local/lib/python2.7/dist-packages/numpy/core/include/numpy/ /usr/local/include/python2.7/numpy

# Build Caffe python bindings

RUN cd /opt/caffe && make pycaffe

{{ if .CPU }}

# Make + run tests

RUN cd /opt/caffe && make test && make runtest

{{ end }}