Merge pull request #9884 from PaddlePaddle/fea/docker_cudnn7

luotao1 · web-flow · commit 6f0df4fd100d · 2018-04-13T10:52:59.000+08:00
fea/docker cudnn7
diff --git a/Dockerfile b/Dockerfile
@@ -1,6 +1,6 @@
 # A image for building paddle binaries
 # Use cuda devel base image for both cpu and gpu environment
-FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04
+FROM nvidia/cuda:8.0-cudnn7-devel-ubuntu16.04
 MAINTAINER PaddlePaddle Authors <paddle-dev@baidu.com>
 
 ARG UBUNTU_MIRROR
