本文档介绍如何使用Docker方式快速安装MindQuantum。首先需要通过Docker方式安装MindSpore,流程介绍在MindSpore的官方网站,以下将重复这部分的内容。
对于CPU后端,可以直接使用以下命令获取最新的稳定镜像:
docker pull swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-cpu:{tag}
其中{tag}为x.y.z或者devel或者runtime,具体请参考MindSpore的docker安装指南
执行以下命令启动Docker容器实例:
docker run -it swr.cn-south-1.myhuaweicloud.com/mindspore/mindspore-cpu:{tag} /bin/bash
其中{tag}与上述tag保持一直。
- 如果你安装的是指定版本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容器
docker exec -it {docker_container} /bin/bash
其中{docker_container} 是docker容器的id或者名字
-
选择编译安装或者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
python -c 'import mindquantum'