335 Star 1.5K Fork 862

MindSpore / docs

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore_ascend_install_docker.md 6.21 KB
一键复制 编辑 原始数据 按行查看 历史
宦晓玲 提交于 2023-07-21 16:52 . modify the md links 1.8

Docker方式安装MindSpore Ascend 910版本

查看源文件

Docker是一个开源的应用容器引擎,让开发者打包他们的应用以及依赖包到一个轻量级、可移植的容器中。通过使用Docker,可以实现MindSpore的快速部署,并与系统环境隔离。

本文档介绍如何在Ascend 910环境的Linux系统上,使用Docker方式快速安装MindSpore。

MindSpore的Docker镜像托管在Huawei SWR上。

目前容器化构建选项支持情况如下:

硬件平台 Docker镜像仓库 标签 说明
Ascend mindspore/mindspore-ascend x.y.z 已经预安装Ascend Data Center Solution 与对应的MindSpore Ascend x.y.z版本的生产环境。

x.y.z对应MindSpore版本号,例如安装1.8.1版本MindSpore时,x.y.z应写为1.8.1。

确认系统环境信息

安装昇腾AI处理器配套软件包

昇腾软件包提供商用版和社区版两种下载途径:

安装包默认安装路径为/usr/local/Ascend。安装后确认当前用户有权限访问昇腾AI处理器配套软件包的安装路径,若无权限,需要root用户将当前用户添加到/usr/local/Ascend所在的用户组。

获取MindSpore镜像

对于Ascend后端,可以直接使用以下命令获取最新的稳定镜像:

docker pull swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-ascend:{tag}

其中:

  • {tag}对应上述表格中的标签。

运行MindSpore镜像

执行以下命令启动Docker容器实例:

docker run -it -u root --ipc=host \
               --device=/dev/davinci0 \
               --device=/dev/davinci1 \
               --device=/dev/davinci2 \
               --device=/dev/davinci3 \
               --device=/dev/davinci4 \
               --device=/dev/davinci5 \
               --device=/dev/davinci6 \
               --device=/dev/davinci7 \
               --device=/dev/davinci_manager \
               --device=/dev/devmm_svm \
               --device=/dev/hisi_hdc \
               -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
               -v /var/log/npu/:/usr/slog \
               swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-ascend:{tag} \
               /bin/bash

其中:

  • {tag}对应上述表格中的标签。

如需使用MindInsight,需设置--network参数为”host”模式, 例如:

docker run -it -u root --ipc=host \
               --network host \
               --device=/dev/davinci0 \
               --device=/dev/davinci1 \
               --device=/dev/davinci2 \
               --device=/dev/davinci3 \
               --device=/dev/davinci4 \
               --device=/dev/davinci5 \
               --device=/dev/davinci6 \
               --device=/dev/davinci7 \
               --device=/dev/davinci_manager \
               --device=/dev/devmm_svm \
               --device=/dev/hisi_hdc \
               -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
               -v /var/log/npu/:/usr/slog \
               swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-ascend:{tag} \
               /bin/bash

验证是否安装成功

按照上述步骤进入MindSpore容器后,测试Docker容器是否正常工作,请运行下面的Python代码并检查输出:

方法一:

python -c "import mindspore;mindspore.run_check()"

如果输出:

MindSpore version: 版本号
The result of multiplication calculation is correct, MindSpore has been installed successfully!

至此,你已经成功通过Docker方式安装了MindSpore Ascend 910版本。

方法二:

import numpy as np
import mindspore as ms
import mindspore.ops as ops

ms.set_context(device_target="Ascend")
x = ms.Tensor(np.ones([1,3,3,4]).astype(np.float32))
y = ms.Tensor(np.ones([1,3,3,4]).astype(np.float32))
print(ops.add(x, y))

代码成功运行时会输出:

[[[[2. 2. 2. 2.]
   [2. 2. 2. 2.]
   [2. 2. 2. 2.]]

  [[2. 2. 2. 2.]
   [2. 2. 2. 2.]
   [2. 2. 2. 2.]]

  [[2. 2. 2. 2.]
   [2. 2. 2. 2.]
   [2. 2. 2. 2.]]]]

至此,你已经成功通过Docker方式安装了MindSpore Ascend 910版本。

验证MindInsight安装:

输入mindinsight start --port 8080, 如提示启动status为success,则安装成功。

升级MindSpore版本

当需要升级MindSpore版本时:

  • 根据需要升级的MindSpore版本,升级对应的Ascend AI处理器配套软件包。
  • 直接使用以下命令获取最新的稳定镜像:
docker pull swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-ascend:{tag}

其中:

  • {tag}对应上述表格中的标签。
1
https://gitee.com/mindspore/docs.git
git@gitee.com:mindspore/docs.git
mindspore
docs
docs
r1.8

搜索帮助

53164aa7 5694891 3bd8fe86 5694891