|
|
@@ -1,4 +1,4 @@ |
|
|
-FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04
|
|
|
+FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04
|
|
|
LABEL maintainer caffe-maint@googlegroups.com
|
|
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
|
@@ -28,7 +28,8 @@ ENV CAFFE_ROOT=/opt/caffe |
|
|
WORKDIR $CAFFE_ROOT
|
|
|
|
|
|
# FIXME: use ARG instead of ENV once DockerHub supports this
|
|
|
-ENV CLONE_TAG=rc4
|
|
|
+# https://github.com/docker/hub-feedback/issues/460
|
|
|
+ENV CLONE_TAG=1.0
|
|
|
|
|
|
RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/BVLC/caffe.git . && \
|
|
|
pip install --upgrade pip && \
|
|
|
|