maxmustermann/nvidia-docker
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add cuDNN v4 RC Dockerfiles based on the CUDA 7.5 image

Browse source
This commit is contained in:
Felix Abecassis 2015-12-10 16:05:05 -08:00
parent b7ee63c0c3
commit af05565e48
5 changed files with 44 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
Makefile
View file

@ -13,7 +13,8 @@ CUDA_LATEST := $(word 1, $(CUDA_VERSIONS))
# cuDNN versions
ifeq ($(OS), ubuntu)
CUDNN_VERSIONS := 7.5-cudnn3-devel 7.5-cudnn3-runtime \
CUDNN_VERSIONS := 7.5-cudnn4-devel 7.5-cudnn4-runtime \
7.5-cudnn3-devel 7.5-cudnn3-runtime \
7.0-cudnn2-devel 7.0-cudnn2-runtime
endif
CUDNN_DEVEL_LATEST := $(word 1, $(CUDNN_VERSIONS))
@ -69,6 +70,12 @@ cudnn-runtime: $(CUDNN_RUNTIME_LATEST)
%-cudnn3-runtime: %-runtime $(OS)/cuda/%/runtime/cudnn3/Dockerfile
docker build -t cuda:$@ $(OS)/cuda/$*/runtime/cudnn3
%-cudnn4-devel: %-devel $(OS)/cuda/%/devel/cudnn4/Dockerfile
docker build -t cuda:$@ $(OS)/cuda/$*/devel/cudnn4
%-cudnn4-runtime: %-runtime $(OS)/cuda/%/runtime/cudnn4/Dockerfile
docker build -t cuda:$@ $(OS)/cuda/$*/runtime/cudnn4
all-cudnn: $(CUDNN_VERSIONS) cudnn cudnn-devel cudnn-runtime
push: all-cuda all-cudnn

2
ubuntu-14.04/cuda/7.5/devel/cudnn
View file

@ -1 +1 @@
cudnn3
cudnn4

17
ubuntu-14.04/cuda/7.5/devel/cudnn4/Dockerfile Normal file
View file

@ -0,0 +1,17 @@
FROM cuda:7.5-devel
MAINTAINER NVIDIA CORPORATION <digits@nvidia.com>
RUN apt-get update && apt-get install -y \
curl && \
rm -rf /var/lib/apt/lists/*
ENV CUDNN_VERSION 4
LABEL com.nvidia.cudnn.version="4"
ENV CUDNN_DOWNLOAD_SUM 4bcc67921018543e9066adb82ffd731f089eac80413e28eab6ae22a8bf5f4aaf
RUN curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v4/cudnn-7.0-linux-x64-v4.0-rc.tgz -O && \
echo "$CUDNN_DOWNLOAD_SUM cudnn-7.0-linux-x64-v4.0-rc.tgz" | sha256sum -c --strict - && \
tar -xzf cudnn-7.0-linux-x64-v4.0-rc.tgz -C /usr/local && \
rm cudnn-7.0-linux-x64-v4.0-rc.tgz && \
ldconfig

2
ubuntu-14.04/cuda/7.5/runtime/cudnn
View file

@ -1 +1 @@
cudnn3
cudnn4

17
ubuntu-14.04/cuda/7.5/runtime/cudnn4/Dockerfile Normal file
View file

@ -0,0 +1,17 @@
FROM cuda:7.5-runtime
MAINTAINER NVIDIA CORPORATION <digits@nvidia.com>
RUN apt-get update && apt-get install -y \
curl && \
rm -rf /var/lib/apt/lists/*
ENV CUDNN_VERSION 4
LABEL com.nvidia.cudnn.version="4"
ENV CUDNN_DOWNLOAD_SUM 4bcc67921018543e9066adb82ffd731f089eac80413e28eab6ae22a8bf5f4aaf
RUN curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v4/cudnn-7.0-linux-x64-v4.0-rc.tgz -O && \
echo "$CUDNN_DOWNLOAD_SUM cudnn-7.0-linux-x64-v4.0-rc.tgz" | sha256sum -c --strict - && \
tar -xzf cudnn-7.0-linux-x64-v4.0-rc.tgz --wildcards 'cuda/lib64/libcudnn.so*' -C /usr/local && \
rm cudnn-7.0-linux-x64-v4.0-rc.tgz && \
ldconfig