diff options
author | Felix Abecassis <fabecassis@nvidia.com> | 2016-08-12 04:08:02 +0300 |
---|---|---|
committer | Felix Abecassis <fabecassis@nvidia.com> | 2016-08-12 04:08:02 +0300 |
commit | 3f9f4b88e482891c0827e622d4ac60504de7034b (patch) | |
tree | 0997ed80682165ecf16ce53705168c4193ce3b94 | |
parent | c1166ae72085ea084f83601e80fc1ee26696c37e (diff) |
Add NVIDIA/caffe 0.15 based on CUDA 7.5
-rw-r--r-- | mk/caffe.mk | 4 | ||||
-rw-r--r-- | ubuntu-14.04/caffe/0.15/Dockerfile | 13 | ||||
-rw-r--r-- | ubuntu-14.04/caffe/Makefile | 3 |
3 files changed, 19 insertions, 1 deletions
diff --git a/mk/caffe.mk b/mk/caffe.mk index 22c141d..4497f5d 100644 --- a/mk/caffe.mk +++ b/mk/caffe.mk @@ -15,6 +15,10 @@ all: $(CAFFE_VERSIONS) latest latest: $(CAFFE_LATEST) $(NV_DOCKER) tag caffe:$< caffe +0.15: $(CURDIR)/0.15/Dockerfile + make -C $(CURDIR)/../cuda 7.5-cudnn5-runtime + $(NV_DOCKER) build -t caffe:$@ $(CURDIR)/$@ + 0.14: $(CURDIR)/0.14/Dockerfile make -C $(CURDIR)/../cuda 7.5-cudnn5-runtime $(NV_DOCKER) build -t caffe:$@ $(CURDIR)/$@ diff --git a/ubuntu-14.04/caffe/0.15/Dockerfile b/ubuntu-14.04/caffe/0.15/Dockerfile new file mode 100644 index 0000000..f48a5df --- /dev/null +++ b/ubuntu-14.04/caffe/0.15/Dockerfile @@ -0,0 +1,13 @@ +FROM cuda:7.5-cudnn5-runtime +MAINTAINER NVIDIA CORPORATION <digits@nvidia.com> + +ENV CAFFE_VERSION 0.15 +LABEL com.nvidia.caffe.version="0.15" + +ENV CAFFE_PKG_VERSION 0.15.9-1+cuda7.5 +RUN apt-get update && apt-get install -y --no-install-recommends --force-yes \ + caffe-nv=$CAFFE_PKG_VERSION \ + caffe-nv-tools=$CAFFE_PKG_VERSION \ + python-caffe-nv=$CAFFE_PKG_VERSION \ + libcaffe-nv0=$CAFFE_PKG_VERSION && \ + rm -rf /var/lib/apt/lists/* diff --git a/ubuntu-14.04/caffe/Makefile b/ubuntu-14.04/caffe/Makefile index a9c1b84..20a1022 100644 --- a/ubuntu-14.04/caffe/Makefile +++ b/ubuntu-14.04/caffe/Makefile @@ -1,5 +1,6 @@ # Copyright (c) 2015-2016, NVIDIA CORPORATION. All rights reserved. -CAFFE_VERSIONS := 0.14 +CAFFE_VERSIONS := 0.15 \ + 0.14 include ../../mk/caffe.mk |