diff options
| author | ngimel <ngimelshein@nvidia.com> | 2017-03-28 17:05:04 -0700 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-03-28 20:05:04 -0400 |
| commit | 215813d7acfcb45d3ff654a14cdd274356b6984d (patch) | |
| tree | 668b8d2f8ece197c5d10d56aec6c379d33237437 /Dockerfile | |
| parent | dc7695a47acac85151d5435ade7bfbcfbbd3e492 (diff) | |
| download | pytorch-215813d7acfcb45d3ff654a14cdd274356b6984d.tar.gz pytorch-215813d7acfcb45d3ff654a14cdd274356b6984d.tar.bz2 pytorch-215813d7acfcb45d3ff654a14cdd274356b6984d.zip | |
Change dockerfile to support for cudnn v6 (#1135)
Diffstat (limited to 'Dockerfile')
| -rw-r--r-- | Dockerfile | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/Dockerfile b/Dockerfile index 1a89da20e2..0f8819616e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,8 @@ -FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu14.04 +FROM nvidia/cuda:8.0-devel-ubuntu16.04 +RUN echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list + +ENV CUDNN_VERSION 6.0.20 RUN apt-get update && apt-get install -y --no-install-recommends \ build-essential \ cmake \ @@ -7,7 +10,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ curl \ ca-certificates \ libjpeg-dev \ - libpng-dev &&\ + libpng-dev \ + libcudnn6=$CUDNN_VERSION-1+cuda8.0 \ + libcudnn6-dev=$CUDNN_VERSION-1+cuda8.0 && \ rm -rf /var/lib/apt/lists/* RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-4.2.12-Linux-x86_64.sh && \ |