安装MindSpore Hub

查看源文件

确认系统环境信息

  • 硬件平台支持Ascend、GPU和CPU。

  • 确认安装Python 3.7.5版本。

  • MindSpore Hub与MindSpore的版本需保持一致。

  • MindSpore Hub支持使用x86 64位或ARM 64位架构的Linux发行版系统。

  • 在联网状态下,安装whl包时会自动下载setup.py中的依赖项,其余情况需自行安装。

安装方式

可以采用pip安装或者源码安装两种方式。

pip安装

下载并安装发布版本列表中的MindSpore Hub whl包。

pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/{version}/Hub/any/mindspore_hub-{version}-py3-none-any.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
  • {version}表示MindSpore Hub版本号,例如下载1.3.0版本MindSpore Hub时,{version}应写为1.3.0。

源码安装

  1. 从Gitee下载源码。

    git clone https://gitee.com/mindspore/hub.git -b r1.5
    
  2. 编译安装MindSpore Hub。

    cd hub
    python setup.py install
    

验证是否成功安装

在能联网的环境中执行以下命令,验证安装结果。

import mindspore_hub as mshub

model = mshub.load("mindspore/cpu/1.0/lenet_v1_mnist", num_class = 10)

如果出现下列提示,说明安装成功:

Downloading data from url https://gitee.com/mindspore/hub/raw/r1.5/mshub_res/assets/mindspore/cpu/1.0/lenet_v1_mnist.md

Download finished!
File size = 0.00 Mb
Checking /home/ma-user/.mscache/mindspore/cpu/1.0/lenet_v1_mnist.md...Passed!