adjust Dockerfile for nvcc support

4 years ago · dd1dfe5bc1
parent 0aa9f900dd
commit dd1dfe5bc1
2 changed files with 2 additions and 2 deletions
--- 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>
--- 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>
`@ -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>`	`MAINTAINER leonwanghui <leon.wanghui@huawei.com>`