MindSpore Graph Learning 是一个基于MindSpore的高效易用的图学习框架。得益于MindSpore的图算融合能力,MindSpore Graph Learning能够针对图模型特有的执行模式进行编译优化,帮助开发者缩短训练时间。 MindSpore Graph Learning 还创新提出了以点为中心编程范式,提供更原生的图神经网络表达方式,并内置覆盖了大部分应用场景的模型,使开发者能够轻松搭建图神经网络。
- 硬件平台确认为Linux系统,暂不支持Windows和Mac。
- 参考MindSpore安装指南,完成MindSpore的安装,要求至少1.6.1版本。
- 其余依赖请参见requirements.txt。
由于MindSpore Graph Learning与MindSpore有依赖关系,请按照根据下表中所指示的对应关系,在MindSpore下载页面下载并安装对应的whl包。
MindSpore Graph Learning 版本 | 分支 | MindSpore运行最低版本 |
---|---|---|
master | master | >=2.0.0 |
0.2.0 | r0.2.0 | >=2.0.0 |
可以采用pip安装或者源码编译安装两种方式。
-
Ascend/CPU
pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.0.0rc1/GraphLearning/cpu/{system_structure}/mindspore_gl-0.2-cp37-cp37m-linux_{system_structure}.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
-
GPU
pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/2.0.0rc1/GraphLearning/gpu/x86_64/cuda-{cuda_verison}/mindspore_gl-0.2-cp37-cp37m-linux_x86_64.whl --trusted-host ms-release.obs.cn-north-4.myhuaweicloud.com -i https://pypi.tuna.tsinghua.edu.cn/simple
- 在联网状态下,安装whl包时会自动下载MindSpore Graph Learning安装包的依赖项(依赖项详情参见requirements.txt),其余情况需自行安装。
{system_structure}
表示为Linux系统架构,可选项为x86_64
和arrch64
。{cuda_verison}
表示为CUDA版本,可选项为10.1
、11.1
和11.6
。
-
从代码仓下载源码
git clone https://gitee.com/mindspore/graphlearning.git
-
编译安装MindSpore Graph Learning
cd graphlearning bash build.sh pip install ./output/mindspore_gl*.whl
执行如下命令,如果没有报错No module named 'mindspore_gl'
,则说明安装成功。
python -c 'import mindspore_gl'