Browse Source

!5941 adjust Dockerfile for nvcc support

Merge pull request !5941 from HW_KK/master
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
4d4b19059e
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      docker/mindspore-gpu/0.7.0-beta/Dockerfile
  2. +1
    -1
      docker/mindspore-gpu/runtime/Dockerfile

+ 1
- 1
docker/mindspore-gpu/0.7.0-beta/Dockerfile View File

@@ -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
- 1
docker/mindspore-gpu/runtime/Dockerfile View File

@@ -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>



Loading…
Cancel
Save