An all-in-one Docker image for deep learning. Contains all the popular DL frameworks (TensorFlow, Theano, Torch, Caffe, etc.)
3.86k
stars
818
forks
source link
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. #101
Open
OneCodeMonkey opened 5 years ago
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`