图神经网络开源框架DGL学习

最新推荐文章于 2024-04-13 09:38:07 发布

湖南大学研究生-杜敏

最新推荐文章于 2024-04-13 09:38:07 发布

阅读量1.5k

点赞数

本文链接：https://blog.csdn.net/qq_20001941/article/details/105341082

版权

本文主要介绍图神经网络开源框架DGL，包括如何通过pip安装、CUDA配置、查阅官方文档以及代码仓库资源。同时，推荐了DGL框架因其简单易用和良好的扩展性。提供了一个视频教程链接和代码示例。

摘要由CSDN通过智能技术生成

图神经网络开源框架DGL学习

还有另外一个开源框架PyTorch Geometric, 在这里不做详细介绍，有兴趣的同学自己去了解一下，个人推荐DGL，简单易用，扩展性好。
个人公众号

图神经网络开源框架DGL_图神经网络,DGL,GNN_ZSCDumin的博客-CSDN博客

1. 使用教程

1.1 pip安装

        pip install dgl           # For CPU Build
pip install dgl-cu90      # For CUDA 9.0 Build
pip install dgl-cu92      # For CUDA 9.2 Build
pip install dgl-cu100     # For CUDA 10.0 Build
pip install dgl-cu101     # For CUDA 10.1 Build

1.2 CUDA安装

        conda install -c dglteam dgl              # For CPU Build
conda install -c dglteam dgl-cuda9.0      # For CUDA 9.0 Build
conda install -c dglteam dgl-cuda10.0     # For CUDA 10.0 Build
conda install -c dglteam dgl-cuda10.1     # For CUDA 10.1 Build

1.3 官方文档

https://docs.dgl.ai/

1.4 代码仓库

https://github.com/dmlc/dgl

2. 视频教程

https://www. bilibili.com/video/BV1J E411K7aT?from=search&seid=15647099632944979039

3. 代码示例

        import dgl
import scipy.io
import urllib.request

data_url = 'https://data.dgl.ai/dataset/ACM.mat'
data_file_path = '/tmp/ACM.mat'

urllib.request.urlretrieve(data_url, data_file_path)
data = scipy.io.loadmat(data_file_path)
print(list(data.keys()))

###############################################################################
# The datase