PyG (PyTorch Geometric) is a library built upon PyTorch to easily write and train Graph Neural Networks (GNNs) for a wide range of applications related to structured data.
- Documentation: PYG DOCUMENTATION
- COLAB NOTEBOOKS AND VIDEO TUTORIALS
- Youtube Channel: Antonio Longa
Import PyG in Colab
pyg-team/pytorch_geometric在Colab导入报错, 可能原因在于cuda版本不匹配, 解决方法见ameya98/install_PyG.py, 代码如下:
# Add this in a Google Colab cell to install the correct version of Pytorch Geometric.
DGL is an easy-to-use, high performance and scalable Python package for deep learning on graphs. DGL is framework agnostic, meaning if a deep graph model is a component of an end-to-end application, the rest of the logics can be implemented in any major frameworks, such as PyTorch, Apache MXNet or TensorFlow.