PyTorch Geometric is a Geometric Deep Learning extension library for PyTorch. In addition, PyTorch Geometric contains the spline-based convolutional layer from our paper:
Matthias Fey, Jan Eric Lenssen, Frank Weichert, Heinrich Müller: SplineCNN: Fast Geometric Deep Learning with Continuous B-Spline Kernels (CVPR 2018)
Install PyTorch manually (Installation).
If cuda is available, check that nvcc is accessible from your terminal, e.g. by typing nvcc --version.
If not, add cuda (/usr/local/cuda/bin) to your $PATH.
Then run:
pip install -r requirements.txt
python setup.py install
cd examples
python cora.py
python setup.py test
Please cite our paper if you use this code in your own work:
@inproceedings{Fey/etal/2018,
title={{SplineCNN}: Fast Geometric Deep Learning with Continuous {B}-Spline Kernels},
author={Fey, Matthias and Lenssen, Jan Eric and Weichert, Frank and M{\"u}ller, Heinrich},
booktitle={IEEE Conference on Computer Vision and Pattern Recognition (CVPR)}
year={2018},
}