安装MindSpore Golden Stick

查看源文件

环境限制

下表列出了安装、编译和运行MindSpore Golden Stick所需的系统环境:

软件名称

版本

Ubuntu

18.04

Python

3.7-3.9

其他的三方依赖请参考requirements文件。 当前MindSpore Golden Stick仅能在Ubuntu18.04上运行。

MindSpore版本依赖关系

MindSpore Golden Stick依赖MindSpore训练推理框架,请按照根据下表中所指示的对应关系,并参考MindSpore安装指导安装对应版本的MindSpore。

MindSpore Golden Stick版本

分支

MindSpore版本

0.4.0

r0.4

2.3.0-rc1

0.3.0

r0.3

2.0.0-rc1, 2.0.0

0.2.0

r0.2

1.9.0

0.1.0

r0.1

1.8.0

安装完MindSpore后,继续安装MindSpore Golden Stick。可以采用pip安装或者源码编译安装两种方式。

pip安装

使用pip命令安装,请从MindSpore Golden Stick下载页面下载并安装whl包。

pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/{ms_version}/GoldenStick/any/mindspore_gs-{mg_version}-py3-none-any.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 在联网状态下,安装whl包时会自动下载MindSpore Golden Stick安装包的依赖项(依赖项详情参见requirement.txt),其余情况需自行安装。

  • {ms_version}表示与MindSpore Golden Stick匹配的MindSpore版本号,例如下载0.4.0版本MindSpore Golden Stick时,{ms_version}应写为2.3.0-rc1。

  • {mg_version}表示MindSpore Golden Stick版本号,例如下载0.4.0版本MindSpore Golden Stick时,{mg_version}应写为0.4.0。

源码编译安装

下载源码,下载后进入golden_stick目录。

bash build.sh
pip install output/mindspore_gs-0.4.0-py3-none-any.whl

其中,build.shgolden_stick目录下的编译脚本文件。

验证安装是否成功

执行以下命令,验证安装结果。导入Python模块不报错即安装成功。

import mindspore_gs