From dd1dfe5bc159a30aa73c3bf28279651aa22dc7a8 Mon Sep 17 00:00:00 2001 From: wuweikang Date: Wed, 9 Sep 2020 11:46:44 +0800 Subject: [PATCH] adjust Dockerfile for nvcc support --- docker/mindspore-gpu/0.7.0-beta/Dockerfile | 2 +- docker/mindspore-gpu/runtime/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/mindspore-gpu/0.7.0-beta/Dockerfile b/docker/mindspore-gpu/0.7.0-beta/Dockerfile index ac6de071d4..839995633e 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 diff --git a/docker/mindspore-gpu/runtime/Dockerfile b/docker/mindspore-gpu/runtime/Dockerfile index 9e8dabe594..64bf5fdb51 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