1.1K Star 1.8K Fork 3.1K

GVPMindSpore / mindquantum

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
install_with_docker.md 2.81 KB
一键复制 编辑 原始数据 按行查看 历史
lvmingfu 提交于 2022-09-26 17:00 . modify code formats for master

MindQuantum Docker安装指南

Docker方式安装MindQuantum

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

获取MindSpore镜像

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

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

其中{tag}为x.y.z或者devel或者runtime,具体请参考MindSpore的docker安装指南

运行MindSpore镜像

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

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

其中{tag}与上述tag保持一直。

验证Mindspore是否安装成功

  • 如果你安装的是指定版本x.y.z的容器,执行如下步骤。

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

import numpy as np
import mindspore as ms
from mindspore import set_context, ops, Tensor

set_context(mode=ms.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.0, 2.0, 2.0, 2.0], [2.0, 2.0, 2.0, 2.0], [2.0, 2.0, 2.0, 2.0]],
        [[2.0, 2.0, 2.0, 2.0], [2.0, 2.0, 2.0, 2.0], [2.0, 2.0, 2.0, 2.0]],
        [[2.0, 2.0, 2.0, 2.0], [2.0, 2.0, 2.0, 2.0], [2.0, 2.0, 2.0, 2.0]],
    ]
]

至此,你已经成功通过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://hiq.huaweicloud.com/download/mindquantum/newest/linux/mindquantum-master-cp37-cp37m-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

验证MindQuantum是否安装成功

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

搜索帮助