mirror of
https://github.com/autistic-symposium/tensorflow-for-deep-learning-py.git
synced 2025-05-10 10:45:04 -04:00
42 lines
No EOL
1.2 KiB
Docker
42 lines
No EOL
1.2 KiB
Docker
FROM ubuntu:14.04
|
|
|
|
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-scipy && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
ENV CAFFE_ROOT=/opt/caffe
|
|
WORKDIR $CAFFE_ROOT
|
|
|
|
# FIXME: clone a specific git tag and use ARG instead of ENV once DockerHub supports this.
|
|
ENV CLONE_TAG=master
|
|
|
|
RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/BVLC/caffe.git . && \
|
|
for req in $(cat python/requirements.txt) pydot; do pip install $req; done && \
|
|
mkdir build && cd build && \
|
|
cmake -DCPU_ONLY=1 .. && \
|
|
make -j"$(nproc)"
|
|
|
|
ENV PYCAFFE_ROOT $CAFFE_ROOT/python
|
|
ENV PYTHONPATH $PYCAFFE_ROOT:$PYTHONPATH
|
|
ENV PATH $CAFFE_ROOT/build/tools:$PYCAFFE_ROOT:$PATH
|
|
RUN echo "$CAFFE_ROOT/build/lib" >> /etc/ld.so.conf.d/caffe.conf && ldconfig
|
|
|
|
WORKDIR /workspace |