A utility script to install PyTorch Geometric with its dependencies, using the pre-built binaries.
MIT License
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.