commit 19c2ad8834c009d1ed5a46590208c76501aea0a1
author Amir Arsalan Soltani <Amir-Arsalan@users.noreply.github.com> Mon Feb 12 14:21:56 2018 -0500
committer Soumith Chintala <soumith@gmail.com> Mon Feb 12 14:21:56 2018 -0500
tree ee54b08fe8398c168c262274db2d6b52e388a4d1
parent 07be53b57f00c58c73bf8f3563213ad77f0c6e15

CUDA 9.0 and cuDNN 7 (#5186)

Dockerfile

diff --git a/Dockerfile b/Dockerfile
index cba8b33..0ea551e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04 
+FROM nvidia/cuda:9.0-cudnn7-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
 
@@ -9,8 +9,8 @@
          curl \
          vim \
          ca-certificates \
-         libnccl2=2.0.5-2+cuda8.0 \
-         libnccl-dev=2.0.5-2+cuda8.0 \
+         libnccl2=2.0.5-2+cuda9.0 \
+         libnccl-dev=2.0.5-2+cuda9.0 \
          libjpeg-dev \
          libpng-dev &&\
      rm -rf /var/lib/apt/lists/*