MindSpore SciAI安装

查看源文件   

确认系统环境信息

安装方式

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

方式一:pip安装

此方式不用克隆源码编译,而是自动下载安装MindSpore官方提供的whl包。

pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.2.0/MindScience/sciai/gpu/{arch}/cuda-11.1/sciai-{version}-cp37-cp37m-linux_{arch}.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 在联网状态下,安装whl包时会自动下载SciAI安装包的依赖项(依赖项详情参见setup.py)。

  • {version}表示SciAI版本号,例如下载0.1.0版本SciAI时,{version}应写为0.1.0

  • {arch}表示系统架构,例如使用的Linux系统是x86架构64位时,{arch}应写为x86_64。如果系统是ARM架构64位,则写为aarch64

下表提供了各架构和Python版本对应的安装命令。

设备

架构

Python

安装命令

Ascend

x86_64

Python=3.7

pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.2.0/MindScience/sciai/gpu/x86_64/cuda-11.1/sciai-0.1.0-cp37-cp37m-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

aarch64

Python=3.7

pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.2.0/MindScience/sciai/ascend/aarch64/sciai-0.1.0-cp37-cp37m-linux_aarch64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

GPU

x86_64

Python=3.7

pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.2.0/MindScience/sciai/gpu/x86_64/cuda-11.1/sciai-0.1.0-cp37-cp37m-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

注意:如果您的conda或python env中已经安装了其他MindScience套件,如MindElecMindFlowMindSponge, 请先卸载环境中的MindScience套件避免pip行为冲突。

方式二:源码安装

  1. 克隆MindScience代码Git仓库。

    cd ~
    git clone https://gitee.com/mindspore/mindscience.git
    
  2. 使用脚本build.sh编译SciAI。

    cd mindscience/SciAI
    bash build.sh -j8
    
  3. 编译完成后,通过如下命令安装编译所得.whl包。

    bash install.sh
    

验证安装

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

python -c 'import sciai'