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

Add cuDNN v3 to the CUDA 7.0 images

Browse source 
Closes: #21
Signed-off-by: Felix Abecassis <fabecassis@nvidia.com>
This commit is contained in:
Evan Lezar 2015-12-11 08:56:58 +01:00 committed by Felix Abecassis
parent 54b3708804
commit c42f9555f1
3 changed files with 35 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
Makefile
View file

@ -16,6 +16,7 @@ ifeq ($(OS), ubuntu)
CUDNN_VERSIONS := 7.5-cudnn4-devel 7.5-cudnn4-runtime \
7.5-cudnn3-devel 7.5-cudnn3-runtime \
7.0-cudnn4-devel 7.0-cudnn4-runtime \
7.0-cudnn3-devel 7.0-cudnn3-runtime \
7.0-cudnn2-devel 7.0-cudnn2-runtime
endif
CUDNN_DEVEL_LATEST := $(word 1, $(CUDNN_VERSIONS))

17
ubuntu-14.04/cuda/7.0/devel/cudnn3/Dockerfile Normal file
View file

@ -0,0 +1,17 @@
FROM cuda:7.0-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 3
LABEL com.nvidia.cudnn.version="3"
ENV CUDNN_DOWNLOAD_SUM 98679d5ec039acfd4d81b8bfdc6a6352d6439e921523ff9909d364e706275c2b
RUN curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v3/cudnn-7.0-linux-x64-v3.0-prod.tgz -O && \
echo "$CUDNN_DOWNLOAD_SUM cudnn-7.0-linux-x64-v3.0-prod.tgz" | sha256sum -c --strict - && \
tar -xzf cudnn-7.0-linux-x64-v3.0-prod.tgz -C /usr/local && \
rm cudnn-7.0-linux-x64-v3.0-prod.tgz && \
ldconfig

17
ubuntu-14.04/cuda/7.0/runtime/cudnn3/Dockerfile Normal file
View file

@ -0,0 +1,17 @@
FROM cuda:7.0-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 3
LABEL com.nvidia.cudnn.version="3"
ENV CUDNN_DOWNLOAD_SUM 98679d5ec039acfd4d81b8bfdc6a6352d6439e921523ff9909d364e706275c2b
RUN curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v3/cudnn-7.0-linux-x64-v3.0-prod.tgz -O && \
echo "$CUDNN_DOWNLOAD_SUM cudnn-7.0-linux-x64-v3.0-prod.tgz" | sha256sum -c --strict - && \
tar -xzf cudnn-7.0-linux-x64-v3.0-prod.tgz --wildcards 'cuda/lib64/libcudnn.so*' -C /usr/local && \
rm cudnn-7.0-linux-x64-v3.0-prod.tgz && \
ldconfig