Dockerfile 382 Bytes
Newer Older
Felix Abecassis US's avatar
Felix Abecassis US committed
1 2 3 4
ARG repository
FROM ${repository}:9.1-runtime-ubuntu16.04
LABEL maintainer "NVIDIA CORPORATION <cudatools@nvidia.com>"

5
ENV CUDNN_VERSION 7.1.2.21
Felix Abecassis US's avatar
Felix Abecassis US committed
6 7 8 9
LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}"

RUN apt-get update && apt-get install -y --no-install-recommends \
            libcudnn7=$CUDNN_VERSION-1+cuda9.1 && \
10
    apt-mark hold libcudnn7 && \
Felix Abecassis US's avatar
Felix Abecassis US committed
11
    rm -rf /var/lib/apt/lists/*