MindSpore安装和训练体验

MindSpore官网建议的安装操作系统为EulerOS和Ubuntu,并不包括Windows。但仔细看了下文档,发现实际上MindSpore是支持Windows平台的,只不过只能以CPU的模式运行(暂时不支持Windows平台的GPU)。如果希望尝试MindSpore对GPU的支持,需要使用Ubuntu系列或者华为的EulerOS。

  1. 进入安装的正题

    安装Anaconda

    非常推荐使用Anaconda作为Python虚拟环境的管理。

    你可以去Anaconda官网下载,一路Next即可。如果遇到问题,可适当参考这里。

    由于Anaconda的安装过程简单直观,这里不详述了。

  2. 创建MindSpore的Python Env

    目前MindSpore仅仅支持Python 3.7.5,下面创建的pyhton env就叫做mindspore:

    conda create -n mindspore python=3.7.5

  3. 安装MindSpore

    下载MindSpore的whl文件,下载地址:点击这里,目前只有一个0.1.0的版本

    mindspore-0.1.0-cp37-cp37m-win_amd64.whl

    在mindspore这个python env里执行:

    conda activate mindspore # 此时你应该在mindspore-0.1.0-cp37-cp37m-win_amd64.whl这个文件的目录里pip install mindspore-0.1.0-cp37-cp37m-win_amd64.whl

    注意,这里很多同学反应安装过程很慢,甚至有失败的情况,基本都是网络惹得祸。因为安装上面的whl文件时会按照一些依赖的libraries,如果你的pip是默认源的话,在国内的网络下载时可能遇到问题的。一个解决办法是重试,另一个办法是把pip源切换为国内的源,比如清华源。

    一个临时的办法可以是:

    pip install -i https://pypi.tuna.tsinghua.edu.cn/simple mindspore-0.1.0-cp37-cp37m-win_amd64.whl

  4. 验证MindSpore是否安装成功

    python

    >>> import mindspore as ms

    >>> print(ms.__version__)

    # 如果成功,上面应该打印版本号

  5. 训练一个LeNet

    华为官方已经在其MindSpore Doc这个Repo里内置了一些examples,比如LeNet。

    直接Fork吧:

    git clone https://github.com/mindspore-ai/docs.git

    仅仅下载lenet.py也可以,路径如下:

    https://github.com/mindspore-ai/docs/blob/master/tutorials/tutorial_code/lenet.py

    因为这个代码里默认使用的训练设备为Ascend,所以需要手动设置一下--device_target为CPU:

    python lenet.py --device_target CPU

    MindSpore用LeNet训练MNIST,大约十几秒就可以跑完一个Epoch了:

    ******Downloading the MNIST dataset******

    MindSpore安装和训练体验_第1张图片

  6. 此时我们已经成功安装了mindspore,并进行了安装验证,跑出来了精度。

你可能感兴趣的:(python,机器学习,深度学习)