mindspore

mindspore-aiApache-2.0活跃维护

华为开源的深度学习训练与推理框架,覆盖移动端、边缘和云场景,主打国产 NPU 生态适配。

MindSpore is a new open source deep learning training/inference framework that could be used for mobile, edge and cloud scenarios.

暂无版本4,723star +4 / 7 天747fork同步于 2026-11-10 17:38 UTC

怎么上手

由 README 提炼,执行前请确认
pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.2.0-rc1/MindSpore/cpu/ubuntu_x86/mindspore-1.2.0rc1-cp37-cp37m-linux_x86_64.whl
硬件要求按 README 说明

README列出支持华为昇腾(Ubuntu/CentOS/EulerOS的x86与aarch64)、GPU CUDA 10.1(Ubuntu-x86)以及CPU(Ubuntu-x86/aarch64、Windows-x86);未给出具体显存要求。

需要 CUDA昇腾 / Ascend有 Docker 镜像
下载:源码 zip·GitHub Releases(安装包)本站不镜像安装包

项目说明

由 README 自动翻译提炼

华为开源深度学习框架,支持端边云训练推理,主打昇腾原生与软硬件协同优化

华为开源的深度学习训练与推理框架,覆盖移动端、边缘和云场景,主打国产 NPU 生态适配。

核心能力

  • 基于源码变换实现自动微分,支持自动控制流
  • 提供自动并行,自动选择数据/模型/混合并行切分策略
  • 原生支持华为昇腾AI处理器,做软硬件协同优化
  • 覆盖移动、边缘、云端多场景训练与推理
  • 提供pip安装、源码编译、Docker镜像多种安装方式

容易踩的坑

pip安装需按硬件平台和操作系统从下载页选对应whl包,不能直接pip install mindspore;示例whl绑定cp37与linux_x86_64,Python版本和架构不匹配会装不上;GPU后端仅列出CUDA 10.1,其他CUDA版本README未说明支持

内容由模型根据 README 生成,可能不完整或不准确,以官方文档为准。看原文 README

最近发版

0 个版本

暂无 release 数据(可能没有发版,或尚未同步)。

用户点评

登录用户的真实使用记录,可按版本和硬件筛选
共 0 条

    还没有人点评。第一个说出真实体验的人最有价值。

    README 原文

    未翻译

    MindSpore Logo

    PyPI - Python Version PyPI Downloads DockerHub LICENSE Slack PRs Welcome

    查看中文

    What Is MindSpore

    MindSpore is a new open source deep learning training/inference framework that could be used for mobile, edge and cloud scenarios. MindSpore is designed to provide development experience with friendly design and efficient execution for the data scientists and algorithmic engineers, native support for Huawei Ascend AI processor, and software hardware co-optimization. At the meantime MindSpore as a global AI open source community, aims to further advance the development and enrichment of the AI software/hardware application ecosystem.

    For more details please check out our Architecture Guide.

    Automatic Differentiation

    Currently, there are two automatic differentiation techniques in mainstream deep learning frameworks:

    • Operator Overloading (OO): Overloading the basic operators of the programming language to encapsulate their gradient rules. Record the operation trajectory of the network during forward execution in an operator overloaded manner, then apply the chain rule to the dynamically generated data flow graph to implement automatic differentiation.
    • Source Transformation (ST): This technology is evolving from the functional programming framework and performs automatic differential transformation on the intermediate expression (the expression form of the program during the compilation process) in the form of just-in-time compilation (JIT), supporting complex control flow scenarios, higher-order functions and closures.

    PyTorch used OO. Compared to ST, OO generates gradient graph in runtime, so it does not need to take function call and control flow into consideration, which makes it easier to develop. However, OO can not perform gradient graph optimization in compilation time and the control flow has to be unfolded in runtime, so it is difficult to achieve extreme optimization in performance.

    MindSpore implemented automatic differentiation based on ST. On the one hand, it supports automatic differentiation of automatic control flow, so it is quite convenient to build models like PyTorch. On the other hand, MindSpore can perform static compilation optimization on neural networks to achieve great performance.

    The implementation of MindSpore automatic differentiation can be understood as the symbolic differentiation of the program itself. Because MindSpore IR is a functional intermediate expression, it has an intuitive correspondence with the composite function in basic algebra. The derivation formula of the composite function composed of arbitrary basic functions can be derived. Each primitive operation in MindSpore IR can correspond to the basic functions in basic algebra, which can build more complex flow control.

    Automatic Parallel

    The goal of MindSpore automatic parallel is to build a training method that combines data parallelism, model parallelism, and hybrid parallelism. It can automatically select a least cost model splitting strategy to achieve automatic distributed parallel training.

    At present, MindSpore uses a fine-grained parallel strategy of splitting operators, that is, each operator in the figure is split into a cluster to complete parallel operations. The splitting strategy during this period may be very complicated, but as a developer advocating Pythonic, you don't need to care about the underlying implementation, as long as the top-level API compute is efficient.

    Installation

    Pip mode method installation

    MindSpore offers build options across multiple backends:

    Hardware PlatformOperating SystemStatus
    Huawei AscendUbuntu-x86✔️
    Ubuntu-aarch64✔️
    EulerOS-aarch64✔️
    CentOS-x86✔️
    CentOS-aarch64✔️
    GPU CUDA 10.1Ubuntu-x86✔️
    CPUUbuntu-x86✔️
    Ubuntu-aarch64✔️
    Windows-x86✔️

    For installation using pip, take CPU and Ubuntu-x86 build version as an example:

    1. Download whl from MindSpore download page, and install the package.

      pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.2.0-rc1/MindSpore/cpu/ubuntu_x86/mindspore-1.2.0rc1-cp37-cp37m-linux_x86_64.whl
      
    2. Run the following command to verify the install.

      import numpy as np
      import mindspore.context as context
      import mindspore.nn as nn
      from mindspore import Tensor
      from mindspore.ops import operations as P
      
      context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
      
      class Mul(nn.Cell):
          def __init__(self):
              super(Mul, self).__init__()
              self.mul = P.Mul()
      
          def construct(self, x, y):
              return self.mul(x, y)
      
      x = Tensor(np.array([1.0, 2.0, 3.0]).astype(np.float32))
      y = Tensor(np.array([4.0, 5.0, 6.0]).astype(np.float32))
      
      mul = Mul()
      print(mul(x, y))
      
      [ 4. 10. 18.]
      

    Use pip mode method to install MindSpore in different environments. Refer to the following documents.

    Source code compilation installation

    Use the source code compilation method to install MindSpore in different environments. Refer to the following documents.

    Docker Image

    MindSpore docker image is hosted on Docker Hub, currently the containerized build options are supported as follows:

    Hardware PlatformDocker Image RepositoryTagDescription
    CPUmindspore/mindspore-cpux.y.zProduction environment with pre-installed MindSpore x.y.z CPU release.
    develDevelopment environment provided to build MindSpore (with CPU backend) from the source, refer to https://www.mindspore.cn/install/en for installation details.
    runtimeRuntime environment provided to install MindSpore binary package with CPU backend.
    GPUmindspore/mindspore-gpux.y.zProduction environment with pre-installed MindSpore x.y.z GPU release.
    develDevelopment environment provided to build MindSpore (with GPU CUDA10.1 backend) from the source, refer to https://www.mindspore.cn/install/en for installation details.
    runtimeRuntime environment provided to install MindSpore binary package with GPU CUDA10.1 backend.

    NOTICE: For GPU devel docker image, it's NOT suggested to directly install the whl package after building from the source, instead we strongly RECOMMEND you transfer and install the whl package inside GPU runtime docker image.

    • CPU

      For CPU backend, you can directly pull and run the latest stable image using the below command:

      docker pull mindspore/mindspore-cpu:1.1.0
      docker run -it mindspore/mindspore-cpu:1.1.0 /bin/bash
      
    • GPU

      For GPU backend, please make sure the nvidia-container-toolkit has been installed in advance, here are some install guidelines for Ubuntu users:

      DISTRIBUTION=$(. /etc/os-release; echo $ID$VERSION_ID)
      curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add -
      curl -s -L https://nvidia.github.io/nvidia-docker/$DISTRIBUTION/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list
      
      sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit nvidia-docker2
      sudo systemctl restart docker
      

      Then edit the file daemon.json:

      $ vim /etc/docker/daemon.json
      {
          "runtimes": {
              "nvidia": {
                  "path": "nvidia-container-runtime",
                  "runtimeArgs": []
              }
          }
      }
      

      Restart docker again:

      sudo systemctl daemon-reload
      sudo systemctl restart docker
      

      Then you can pull and run the latest stable image using the below command:

      docker pull mindspore/mindspore-gpu:1.1.0
      docker run -it -v /dev/shm:/dev/shm --runtime=nvidia --privileged=true mindspore/mindspore-gpu:1.1.0 /bin/bash
      

      To test if the docker image works, please execute the python code below and check the output:

      import numpy as np
      import mindspore.context as context
      from mindspore import Tensor
      from mindspore.ops import functional as F
      
      context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
      
      x = Tensor(np.ones([1,3,3,4]).astype(np.float32))
      y = Tensor(np.ones([1,3,3,4]).astype(np.float32))
      print(F.tensor_add(x, y))
      
      [[[ 2.  2.  2.  2.],
      [ 2.  2.  2.  2.],
      [ 2.  2.  2.  2.]],
      
      [[ 2.  2.  2.  2.],
      [ 2.  2.  2.  2.],
      [ 2.  2.  2.  2.]],
      
      [[ 2.  2.  2.  2.],
      [ 2.  2.  2.  2.],
      [ 2.  2.  2.  2.]]]
      

    If you want to learn more about the building process of MindSpore docker images, please check out docker repo for the details.

    Quickstart

    See the Quick Start to implement the image classification.

    Docs

    More details about installation guide, tutorials and APIs, please see the User Documentation.

    Community

    Governance

    Check out how MindSpore Open Governance works.

    Communication

    Contributing

    Welcome contributions. See our Contributor Wiki for more details.

    Maintenance phases

    Project stable branches will be in one of the following states:

    StateTime frameSummary
    Planning1 - 3 monthsFeatures are under planning.
    Development3 monthsFeatures are under development.
    Maintained6 - 12 monthsAll bugfixes are appropriate. Releases produced.
    Unmaintained0 - 3 monthsAll bugfixes are appropriate. No Maintainers and No Releases produced.
    End Of Life (EOL)N/ABranch no longer accepting changes.

    Maintenance status

    BranchStatusInitial Release DateNext PhaseEOL Date
    r2.2Maintained2023-10-18Unmaintained 2024-10-18 estimated
    r2.1Maintained2023-07-29Unmaintained 2024-07-29 estimated
    r2.0Maintained2023-06-15Unmaintained 2024-06-15 estimated
    r1.10End Of Life2023-02-022024-02-02
    r1.9End Of Life2022-10-262023-10-26
    r1.8End Of Life2022-07-292023-07-29
    r1.7End Of Life2022-04-292023-04-29
    r1.6End Of Life2022-01-292023-01-29
    r1.5End Of Life2021-10-152022-10-15
    r1.4End Of Life2021-08-152022-08-15
    r1.3End Of Life2021-07-152022-07-15
    r1.2End Of Life2021-04-152022-04-29
    r1.1End Of Life2020-12-312021-09-30
    r1.0End Of Life2020-09-242021-07-30
    r0.7End Of Life2020-08-312021-02-28
    r0.6End Of Life2020-07-312020-12-30
    r0.5End Of Life2020-06-302021-06-30
    r0.3End Of Life2020-05-312020-09-30
    r0.2End Of Life2020-04-302020-08-31
    r0.1End Of Life2020-03-282020-06-30

    Release Notes

    The release notes, see our RELEASE.

    License

    Apache License 2.0