安装MindSpore Flow

  

确认系统环境信息

• 硬件平台为Ascend、GPU。

• 参考MindSpore安装指南，完成MindSpore的安装。

• 其余依赖请参见requirements.txt。

• MindSpore Flow需MindSpore版本>=2.0.0(建议2.2.10)，Python版本需>=3.7。

安装方式

可以采用pip安装或者源码编译安装两种方式。

pip安装

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

源码安装

1. 从Gitee下载源码。

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

2. 编译Ascend后端源码。

   bash build.sh -e ascend -j8
   

3. 编译GPU后端源码。

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

4. 安装编译所得whl包。

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

验证是否成功安装

执行如下命令，如果没有报错No module named 'mindflow'，则说明安装成功。

python -c 'import mindflow'