Go to file
义峰潘 2bef22d8a3 fix bprop cache caused error with variable params 2020-04-10 12:21:40 +08:00
.gitee Add issue and PR templates for gitee and github 2020-03-30 20:17:13 +08:00
.github Add issue and PR templates for gitee and github 2020-03-30 20:17:13 +08:00
cmake add custom environment variable ASCEND_CUSTOM_PATH for customized Ascend software installation 2020-04-09 19:07:01 +08:00
config fix bug when dump op with trans_flag true 2020-04-01 16:21:00 +08:00
docker Update some required system packages in dockerfile 2020-04-03 16:17:11 +08:00
docs fix typo 2020-03-31 12:01:58 +08:00
example modify comment 2020-04-09 09:23:39 +08:00
graphengine@5bd0dc1ed5 add custom environment variable ASCEND_CUSTOM_PATH for customized Ascend software installation 2020-04-09 19:07:01 +08:00
mindspore fix bprop cache caused error with variable params 2020-04-10 12:21:40 +08:00
predict initial version 2020-03-27 22:54:54 +08:00
scripts initial version 2020-03-27 22:54:54 +08:00
tests fix bprop cache caused error with variable params 2020-04-10 12:21:40 +08:00
third_party initial version 2020-03-27 22:54:54 +08:00
.clang-format initial version 2020-03-27 22:54:54 +08:00
.gitignore initial version 2020-03-27 22:54:54 +08:00
.gitmodules initial version 2020-03-27 22:54:54 +08:00
CMakeLists.txt fix ME+GE compile error 2020-04-03 14:59:33 +08:00
CONTRIBUTING.md Update setuptool info 2020-04-09 15:58:42 +08:00
LICENSE initial version 2020-03-27 22:54:54 +08:00
NOTICE initial version 2020-03-27 22:54:54 +08:00
README.md Update setuptool info 2020-04-09 15:58:42 +08:00
RELEASE.md Update setuptool info 2020-04-09 15:58:42 +08:00
SECURITY.md initial version 2020-03-27 22:54:54 +08:00
Third_Party_Open_Source_Software_Notice updata mkl-dnn link and md5 2020-04-07 16:27:44 +08:00
autogen.sh initial version 2020-03-27 22:54:54 +08:00
build.sh !145 Distinguish package name according to hardware platform in build.sh 2020-04-09 12:01:24 +08:00
dbg_dump_parser.sh initial version 2020-03-27 22:54:54 +08:00
package.sh Update setuptool info 2020-04-09 15:58:42 +08:00
py_filter initial version 2020-03-27 22:54:54 +08:00
requirements.txt initial version 2020-03-27 22:54:54 +08:00
setup.py Update setuptool info 2020-04-09 15:58:42 +08:00

README.md

MindSpore Logo

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 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.

MindSpore Architecture

For more details please check out our Architecture Guide.

Automatic Differentiation

There are currently three automatic differentiation techniques in mainstream deep learning frameworks:

  • Conversion based on static compute graph: Convert the network into a static data flow graph at compile time, then turn the chain rule into a data flow graph to implement automatic differentiation.
  • Conversion based on dynamic compute graph: 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.
  • Conversion based on source code: 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.

TensorFlow adopted static calculation diagrams in the early days, whereas PyTorch used dynamic calculation diagrams. Static maps can utilize static compilation technology to optimize network performance, however, building a network or debugging it is very complicated. The use of dynamic graphics is very convenient, but it is difficult to achieve extreme optimization in performance.

But MindSpore finds another way, automatic differentiation based on source code conversion. 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.

Automatic Differentiation

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.

Automatic Parallel

At present, MindSpore uses a fine-grained parallel strategy of splitting operators, that is, each operator in the figure is splited 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

Binaries

MindSpore offers build options across multiple backends:

Hardware Platform Operating System Status
Ascend910 Ubuntu-x86 ✔️
EulerOS-x86 ✔️
EulerOS-aarch64 ✔️
GPU CUDA 9.2 Ubuntu-x86 ✔️
GPU CUDA 10.1 Ubuntu-x86 ✔️
CPU Ubuntu-x86 ✔️

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

  1. Download whl from MindSpore website, and install the package.

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

    python -c 'import mindspore'
    

From Source

Install MindSpore.

Docker Image

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

Hardware Platform Docker Image URL
CPU mindspore/mindspore-cpu:0.1.0-alpha
GPU CUDA 9.2 mindspore/mindspore-cuda9.2:0.1.0-alpha
GPU CUDA 10.1 mindspore/mindspore-cuda10.1:0.1.0-alpha
Ascend

Take CPU for example, you can directly pull the image using the below command:

docker pull mindspore/mindspore-cpu:0.1.0-alpha

If anyone wants to learn more about the build process of MindSpore docker images, please check out docker folder 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.

Release Notes

The release notes, see our RELEASE.

License

Apache License 2.0