329 Star 1.4K Fork 678

MindSpore / docs

加入 Gitee
与超过 800 万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore_cpu_install_source.md 4.84 KB
一键复制 编辑 Web IDE 原始数据 按行查看 历史

源码编译方式安装MindSpore CPU版本

本文档介绍如何在CPU环境的Linux系统上,使用源码编译方式快速安装MindSpore。

详细步骤可以参考社区提供的实践——在Ubuntu(CPU)上进行源码编译安装MindSpore,在此感谢社区成员damon0626的分享。

确认系统环境信息

  • 确认安装64位操作系统,其中Ubuntu 18.04是经过验证的。

  • 确认安装GCC 7.3.0版本

  • 确认安装gmp 6.1.2版本

  • 确认安装Python 3.7.5或3.9.0版本。如果未安装或者已安装其他版本的Python,可以选择下载并安装:

  • 确认安装CMake 3.18.3及以上版本

    • 安装完成后需将CMake所在路径添加到系统环境变量。
  • 确认安装wheel 0.32.0及以上版本

  • 确认安装patch 2.5及以上版本

    • 安装完成后需将patch所在路径添加到系统环境变量中。
  • 确认安装NUMA 2.0.11及以上版本。 如果未安装,使用如下命令下载安装:

    apt-get install libnuma-dev
  • 确认安装git工具。 如果未安装,使用如下命令下载安装:

    apt-get install git # for linux distributions using apt, e.g. ubuntu
    yum install git     # for linux distributions using yum, e.g. centos

从代码仓下载源码

git clone https://gitee.com/mindspore/mindspore.git -b r1.5

编译MindSpore

在源码根目录下执行如下命令。

bash build.sh -e cpu -j4

其中:
如果编译机性能较好,可在执行中增加-j{线程数}来增加线程数量。如bash build.sh -e cpu -j12

安装MindSpore

pip install output/mindspore-{version}-{python_version}-linux_{arch}.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

其中:

  • 在联网状态下,安装whl包时会自动下载mindspore安装包的依赖项(依赖项详情参见setup.py中的required_package),其余情况需自行安装。运行模型时,需要根据ModelZoo中不同模型指定的requirements.txt安装额外依赖,常见依赖可以参考requirements.txt
  • {version}表示MindSpore版本号,例如安装1.5.0-rc1版本MindSpore时,{version}应写为1.5.0rc1。
  • {arch}表示系统架构,例如使用的Linux系统是x86架构64位时,{arch}应写为x86_64。如果系统是ARMv8架构64位,则写为aarch64
  • {python_version}表示用户的Python版本,Python版本为3.7.5时,{python_version}应写为cp37-cp37m。Python版本为3.9.0时,则写为cp39-cp39

验证安装是否成功

python -c "import mindspore;mindspore.run_check()"

如果输出:

MindSpore version: 版本号
The result of multiplication calculation is correct, MindSpore has been installed successfully!

说明MindSpore安装成功了。

升级MindSpore版本

当需要升级MindSpore版本时,可执行如下命令:

  • 直接在线升级

    pip install --upgrade mindspore
  • 本地源码编译升级

    在源码根目录下执行编译脚本build.sh成功后,在output目录下找到编译生成的whl安装包,然后执行命令进行升级。

    pip install --upgrade mindspore-{version}-{python_version}-linux_{arch}.whl
1
https://gitee.com/mindspore/docs.git
git@gitee.com:mindspore/docs.git
mindspore
docs
docs
r1.5

搜索帮助