安装MindSpore Earth
确认系统环境信息
硬件平台为Ascend、GPU。
参考MindSpore安装指南,完成MindSpore的安装。
其余依赖请参见requirements.txt。
安装方式
可以采用pip安装或者源码编译安装两种方式。
pip安装
# gpu and ascend are supported
export DEVICE_NAME=gpu
pip install mindearth_${DEVICE_NAME}
源码安装
从Gitee下载源码。
git clone https://gitee.com/mindspore/mindscience.git cd {PATH}/mindscience/MindEarth
编译Ascend后端源码。
bash build.sh -e ascend -j8
编译GPU后端源码。
export CUDA_PATH={your_cuda_path} bash build.sh -e GPU -j8
安装编译所得whl包。
cd {PATH}/mindscience/MindEarth/output pip install mindearth_*.whl
验证是否成功安装
执行如下命令,如果没有报错No module named 'mindearth'
,则说明安装成功。
python -c 'import mindearth'