MindSpore Flow Installation

View Source On Gitee  

System Environment Information Confirmation

  • The hardware platform should be Ascend, GPU.

  • See our MindSpore Installation Guide to install MindSpore.

  • All other dependencies are included in requirements.txt.

  • MindSpore Flow requires MindSpore version >=2.0.0(Recommended 2.2.10), and Python version requires >=3.7.

Installation

You can install MindSpore Flow either by pip or by source code.

Installation by pip

# gpu and ascend are supported
export DEVICE_NAME=gpu
pip install mindflow_${DEVICE_NAME}

Installation by Source Code

  1. Download source code from Gitee.

    git clone https://gitee.com/mindspore/mindscience.git
cd {PATH}/mindscience/MindFlow

  2. Compile in Ascend backend.

    bash build.sh -e ascend -j8

  3. Compile in GPU backend.

    export CUDA_PATH={your_cuda_path}
bash build.sh -e GPU -j8

  4. Install the compiled .whl file.

    cd {PATH}/mindscience/MindFLow/output
pip install mindflow_*.whl

Installation Verification

Successfully installed, if there is no error message such as No module named 'mindflow' when execute the following command:

python -c 'import mindflow'