1 Star 0 Fork 3.3K

youryanmi / mindquantum

forked from MindSpore / mindquantum 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
install_with_docker.md 2.86 KB
一键复制 编辑 原始数据 按行查看 历史
donghufeng 提交于 2021-10-20 11:08 . HiQ 5.0

MindQuantum Docker安装指南

Docker方式安装MindQuantum

本文档介绍如何使用Docker方式快速安装MindQuantum。首先需要通过Docker方式安装MindSpore,流程介绍在MindSpore的官方网站,以下将重复这部分的内容。

获取MindSpore镜像

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

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

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

运行MindSpore镜像

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

docker run -it swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-cpu:{tag} /bin/bash

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

验证Mindspore是否安装成功

  • 如果你安装的是指定版本x.y.z的容器。

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

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

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

x = Tensor(np.ones([1,3,3,4]).astype(np.float32))
y = 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 CPU版本。

在Docker容器内安装Mindquantum

  1. 进入Docker容器

    docker exec -it {docker_container} /bin/bash

    其中{docker_container} 是docker容器的id或者名字

  2. 选择编译安装或者pip安装

    编译安装:

    git clone https://gitee.com/mindspore/mindquantum.git
    cd ~/mindquantum
    python setup.py install --user

    pip安装:

    pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.2.0-rc1/MindQuantum/ubuntu_x86/mindquantum-0.1.0-py3-none-any.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple

验证MindQuantum是否安装成功

python -c 'import mindquantum'
Python
1
https://gitee.com/youyanmi/mindquantum.git
git@gitee.com:youyanmi/mindquantum.git
youyanmi
mindquantum
mindquantum
master

搜索帮助