首页 > 其他分享> 文章详细

深度学习框架 MindSpore —— 华为出品的AI计算框架， docker 安装

2021-07-04 21:03:29 阅读：179 来源： 互联网

标签：框架 AI -- https nvidia docker com mindspore

深度学习框架 MindSpore —— 华为出品的AI计算框架

官网地址：

https://www.mindspore.cn/

源代码地址：

https://gitee.com/devilmaycry812839668/mindspore

==============================================================

安装方法：

（注意：这里假设已经安装好了docker及NVIDIA-docker）

官网给出的 nvidia-container-toolkit 安装方式：

# Acquire version of operating system version
DISTRIBUTION=$(. /etc/os-release; echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$DISTRIBUTION/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit nvidia-docker2
sudo systemctl restart docker

docker以及 NVIDIA-docker的安装也可以参考：

https://www.cnblogs.com/devilmaycry812839668/p/13396076.html

https://www.cnblogs.com/devilmaycry812839668/p/13691236.html

https://www.cnblogs.com/devilmaycry812839668/p/13686177.html

https://www.cnblogs.com/devilmaycry812839668/p/13704389.html

https://www.cnblogs.com/devilmaycry812839668/p/13748410.html

再次重启Docker:

sudo systemctl daemon-reload
sudo systemctl restart docker

获取MindSpore镜像

对于GPU后端，可以直接使用以下命令获取最新的稳定镜像：

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

这里到今晚位置只能是：

获取MindSpore镜像

对于GPU后端，可以直接使用以下命令获取最新的稳定镜像：

docker pull swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu:1.1.1

运行MindSpore镜像

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

docker run -it -v /dev/shm:/dev/shm --runtime=nvidia --privileged=true swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu:{tag} /bin/bash

其中：

-v /dev/shm:/dev/shm 将NCCL共享内存段所在目录挂载至容器内部；
--runtime=nvidia 用于指定容器运行时为nvidia-container-runtime；
--privileged=true 赋予容器扩展的能力;
{tag}对应上述表格中的标签。

不过，个人建议启动容器的方式如下：

sudo docker run -it -v /dev/shm:/dev/shm -v /home/devil/data:/data -p 8000:22 -p 8001:8001 -p 8002:8002 -p 8003:8003 -p 8004:8004 -p 8005:8005  --runtime=nvidia --privileged=true swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu:1.1.1 /bin/bash

sudo docker run -it -v /dev/shm:/dev/shm -v /home/devil/data:/data -p 8000:22 -p 8001:8001 -p 8002:8002 -p 8003:8003 -p 8004:8004 -p 8005:8005 --runtime=nvidia --privileged=true swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-gpu:1.1.1 /bin/bash

配置好多个文件映射及网络端口映射。

验证是否安装成功：

import numpy as np
import mindspore.context as context
import mindspore.ops as ops
from mindspore import Tensor

context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")

x = Tensor(np.ones([1,3,3,4]).astype(np.float32))
y = Tensor(np.ones([1,3,3,4]).astype(np.float32))
print(ops.tensor_add(x, y))

标签：框架,AI,--,https,nvidia,docker,com,mindspore
来源： https://www.cnblogs.com/devilmaycry812839668/p/14969877.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

深度学习框架 MindSpore —— 华为出品的AI计算框架， docker 安装

获取MindSpore镜像

获取MindSpore镜像

运行MindSpore镜像