本文是在宿主机Ubuntu16.04上安装Docker(nvidia-docker),并成功进行MindSpore1.0.0的GPU训练;
- Ubuntu 16.04
- Docker
- Nvidia-docker (cuda:10.1-base)----可使用GPU的docker容器
- mindspore-gpu:1.0.0 Dokcer hub上配置好的mindspore-gpu镜像
一、 先讲述成功的流程
【1】mindspore官网GPU安装教程查看具体流程
- 系统需要ubuntu18.04,但是我的ubuntu系统版本是16.04,因为是多人共用系统,不能直接升级系统,只能通过安装docker的方式;
cat /proc/version # 使用此命令查看系统版本
【2】安装Docker—我的系统已安装docker
这里提供一篇参考文章: