Merge pull request #3902 from mrteera/master

Fix Dockerfile to be able to train mnist with GPU
chainer · Dec 15, 2017 · 687593f · 687593f
2 parents 1cff9ae + 3f92398
commit 687593f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/docker/python2/Dockerfile b/docker/python2/Dockerfile
@@ -1,9 +1,11 @@
-FROM nvidia/cuda:7.5-cudnn5-devel
+FROM nvidia/cuda:8.0-cudnn6-devel
 
 RUN apt-get update -y && \
     apt-get install -y --no-install-recommends \
     python-dev \
-    python-pip && \
+    python-pip \
+    python-wheel \
+    python-setuptools && \
     rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
 
 RUN pip install cupy==4.0.0b2 chainer==4.0.0b2
diff --git a/docker/python3/Dockerfile b/docker/python3/Dockerfile
@@ -1,9 +1,11 @@
-FROM nvidia/cuda:7.5-cudnn5-devel
+FROM nvidia/cuda:8.0-cudnn6-devel
 
 RUN apt-get update -y && \
     apt-get install -y --no-install-recommends \
     python3-dev \
-    python3-pip && \
+    python3-pip \
+    python3-setuptools \
+    python3-wheel && \
     rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
 
 RUN pip3 install cupy==4.0.0b2 chainer==4.0.0b2