mberr / pytorch-geometric-installer

A utility script to install PyTorch Geometric with its dependencies, using the pre-built binaries.

PyTorch Geometric Install Helper

Update: PyTorch Geometric now has a similar tool in their documention.

Generate pip install commands for installation of PyTorch Geometric as described here, e.g.

$ python generate_pip.py
# GPU available: True
# Versions
# Torch: 1.10.0+cu113
# CUDA : 11.3
# CudNN: 8200

    
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
pip install torch-cluster -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
pip install torch-spline-conv -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
pip install torch-geometric

Requires PyTorch to be installed first.

Note: If the used PyTorch version was compiled for CUDA, but CUDA is not available for torch, it will install the cpu version of PyTorch Geometric.