floydhub / dl-docker

An all-in-one Docker image for deep learning. Contains all the popular DL frameworks (TensorFlow, Theano, Torch, Caffe, etc.)

Home Page:https://www.floydhub.com

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

If you failed to install caffe, I think you can drop it if we don't use it. Thus at least we can use it.

OneCodeMonkey opened this issue · comments

commented
commented

The solution in PR are with GPU mode, I use in CPU mode then I can't solve this problem.
Coz I 'm not going to use this extension(caffe), I drop it.

Dockerfile.cpu:
## Install Caffe
#RUN git clone -b ${CAFFE_VERSION} --depth 1 https://github.com/BVLC/caffe.git /root/caffe && \
# cd /root/caffe && \
# cat python/requirements.txt | xargs -n1 pip install && \
# mkdir build && cd build && \
# cmake -DCPU_ONLY=1 -DBLAS=Open .. && \
# make -j"$(nproc)" all && \
# make install# >

## Set up Caffe environment variables
#ENV CAFFE_ROOT=/root/caffe
#ENV PYCAFFE_ROOT=$CAFFE_ROOT/python
#ENV PYTHONPATH=$PYCAFFE_ROOT:$PYTHONPATH \
# PATH=$CAFFE_ROOT/build/tools:$PYCAFFE_ROOT:$PATH# >

#RUN echo "$CAFFE_ROOT/build/lib" >> /etc/ld.so.conf.d/caffe.conf && ldconfig