README.md 10.6 KB
Newer Older
Z
zhunaipan 已提交
1 2 3
![MindSpore Logo](docs/MindSpore-logo.png "MindSpore logo")
============================================================

4 5 6
[查看中文](./README_CN.md)

- [What Is MindSpore](#what-is-mindspore)
Z
zhunaipan 已提交
7 8 9 10 11
    - [Automatic Differentiation](#automatic-differentiation)
    - [Automatic Parallel](#automatic-parallel)
- [Installation](#installation)
    - [Binaries](#binaries)
    - [From Source](#from-source)
12
    - [Docker Image](#docker-image)
Z
zhunaipan 已提交
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
- [Quickstart](#quickstart)
- [Docs](#docs)
- [Community](#community)
    - [Governance](#governance)
    - [Communication](#communication)
- [Contributing](#contributing)
- [Release Notes](#release-notes)
- [License](#license)

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

<img src="docs/MindSpore-architecture.png" alt="MindSpore Architecture" width="600"/>

L
leiyuning 已提交
34
For more details please check out our [Architecture Guide](https://www.mindspore.cn/docs/en/master/architecture.html).
Z
zhunaipan 已提交
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57

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

<img src="docs/Automatic-differentiation.png" alt="Automatic Differentiation" width="600"/>

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.

<img src="docs/Automatic-parallel.png" alt="Automatic Parallel" width="600"/>

58
At present, MindSpore uses a fine-grained parallel strategy of splitting operators, that is, each operator in the figure is splitted 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.
Z
zhunaipan 已提交
59 60 61 62 63 64 65 66 67 68 69 70 71 72

## Installation

### Binaries

MindSpore offers build options across multiple backends:

| Hardware Platform | Operating System | Status |
| :---------------- | :--------------- | :----- |
| Ascend910 | Ubuntu-x86 | ✔️ |
|  | EulerOS-x86 | ✔️ |
|  | EulerOS-aarch64 | ✔️ |
| GPU CUDA 10.1 | Ubuntu-x86 | ✔️ |
| CPU | Ubuntu-x86 | ✔️ |
L
leonwanghui 已提交
73
|  | Ubuntu-aarch64 | ✔️ |
74
|  | Windows-x86 | ✔️ |
Z
zhunaipan 已提交
75

L
leonwanghui 已提交
76
For installation using `pip`, take `CPU` and `Ubuntu-x86` build version as an example:
Z
zhunaipan 已提交
77

78
1. Download whl from [MindSpore download page](https://www.mindspore.cn/versions/en), and install the package.
Z
zhunaipan 已提交
79 80

    ```
J
jonyguo 已提交
81
    pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/0.7.0-beta/MindSpore/cpu/ubuntu_x86/mindspore-0.7.0-cp37-cp37m-linux_x86_64.whl
Z
zhunaipan 已提交
82 83 84 85
    ```

2. Run the following command to verify the install.

86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
    ```python
    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))
    ```
Z
zhunaipan 已提交
109
    ```
110
    [ 4. 10. 18.]
Z
zhunaipan 已提交
111 112 113 114 115 116
    ```

### From Source

[Install MindSpore](https://www.mindspore.cn/install/en).

117 118 119 120 121
### Docker Image

MindSpore docker image is hosted on [Docker Hub](https://hub.docker.com/r/mindspore),
currently the containerized build options are supported as follows:

122 123
| Hardware Platform | Docker Image Repository | Tag | Description |
| :---------------- | :---------------------- | :-- | :---------- |
124
| CPU | `mindspore/mindspore-cpu` | `x.y.z` | Production environment with pre-installed MindSpore `x.y.z` CPU release. |
125
|  |  | `devel` | Development environment provided to build MindSpore (with `CPU` backend) from the source, refer to https://www.mindspore.cn/install/en for installation details. |
L
leonwanghui 已提交
126
|  |  | `runtime` | Runtime environment provided to install MindSpore binary package with `CPU` backend. |
127
| GPU | `mindspore/mindspore-gpu` | `x.y.z` | Production environment with pre-installed MindSpore `x.y.z` GPU release. |
128
|  |  | `devel` | Development environment provided to build MindSpore (with `GPU CUDA10.1` backend) from the source, refer to https://www.mindspore.cn/install/en for installation details. |
129
|  |  | `runtime` | Runtime environment provided to install MindSpore binary package with `GPU CUDA10.1` backend. |
L
leonwanghui 已提交
130
| Ascend | <center>&mdash;</center> | <center>&mdash;</center> | Coming soon. |
131

132 133
> **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.

134 135
* CPU

136
    For `CPU` backend, you can directly pull and run the latest stable image using the below command:
137
    ```
J
jonyguo 已提交
138 139
    docker pull mindspore/mindspore-cpu:0.7.0-beta
    docker run -it mindspore/mindspore-cpu:0.7.0-beta /bin/bash
140 141 142 143
    ```

* GPU

L
leonwanghui 已提交
144
    For `GPU` backend, please make sure the `nvidia-container-toolkit` has been installed in advance, here are some install guidelines for `Ubuntu` users:
145 146 147 148 149 150 151 152
    ```
    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
    ```
153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169
    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
    ```
170
    Then you can pull and run the latest stable image using the below command:
171
    ```
J
jonyguo 已提交
172 173
    docker pull mindspore/mindspore-gpu:0.7.0-beta
    docker run -it --runtime=nvidia --privileged=true mindspore/mindspore-gpu:0.7.0-beta /bin/bash
174 175 176 177 178
    ```

    To test if the docker image works, please execute the python code below and check the output:
    ```python
    import numpy as np
179
    import mindspore.context as context
180 181 182
    from mindspore import Tensor
    from mindspore.ops import functional as F

L
leonwanghui 已提交
183
    context.set_context(mode=context.PYNATIVE_MODE, device_target="GPU")
184

185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201
    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.]]]
    ```
202

L
leonwanghui 已提交
203
If you want to learn more about the building process of MindSpore docker images,
204
please check out [docker](docker/README.md) repo for the details.
205

Z
zhunaipan 已提交
206 207
## Quickstart

L
leiyuning 已提交
208
See the [Quick Start](https://www.mindspore.cn/tutorial/en/master/quick_start/quick_start.html)
Z
zhunaipan 已提交
209 210 211 212 213 214 215 216 217 218 219 220 221 222 223
to implement the image classification.

## Docs

More details about installation guide, tutorials and APIs, please see the
[User Documentation](https://gitee.com/mindspore/docs).

## Community

### Governance

Check out how MindSpore Open Governance [works](https://gitee.com/mindspore/community/blob/master/governance.md).

### Communication

L
leonwanghui 已提交
224
- [MindSpore Slack](https://join.slack.com/t/mindspore/shared_invite/zt-dgk65rli-3ex4xvS4wHX7UDmsQmfu8w) - Communication platform for developers.
Z
zhunaipan 已提交
225
- IRC channel at `#mindspore` (only for meeting minutes logging purpose)
L
leonwanghui 已提交
226 227
- Video Conferencing: TBD
- Mailing-list: <https://mailweb.mindspore.cn/postorius/lists>
Z
zhunaipan 已提交
228 229 230 231 232 233 234 235 236 237 238 239 240

## Contributing

Welcome contributions. See our [Contributor Wiki](CONTRIBUTING.md) for
more details.

## Release Notes

The release notes, see our [RELEASE](RELEASE.md).

## License

[Apache License 2.0](LICENSE)