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

5
RUN echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list
Felix Abecassis US's avatar
Felix Abecassis US committed
6

7
ENV CUDNN_VERSION 6.0.21
Felix Abecassis US's avatar
Felix Abecassis US committed
8 9 10 11 12
LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}"

RUN apt-get update && apt-get install -y --no-install-recommends \
            libcudnn6=$CUDNN_VERSION-1+cuda8.0 && \
    rm -rf /var/lib/apt/lists/*