diff --git a/docker/mindspore-gpu/0.7.0-beta/Dockerfile b/docker/mindspore-gpu/0.7.0-beta/Dockerfile
index ac6de071d41..839995633e3 100644
--- a/docker/mindspore-gpu/0.7.0-beta/Dockerfile
+++ b/docker/mindspore-gpu/0.7.0-beta/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04
+FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
 
 MAINTAINER leonwanghui <leon.wanghui@huawei.com>
 
diff --git a/docker/mindspore-gpu/runtime/Dockerfile b/docker/mindspore-gpu/runtime/Dockerfile
index 9e8dabe594b..64bf5fdb51d 100644
--- a/docker/mindspore-gpu/runtime/Dockerfile
+++ b/docker/mindspore-gpu/runtime/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04
+FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
 
 MAINTAINER leonwanghui <leon.wanghui@huawei.com>