https://github.com/mberr/pytorch-geometric-installer
A utility script to install PyTorch Geometric with its dependencies, using the pre-built binaries.
https://github.com/mberr/pytorch-geometric-installer
pytorch pytorch-geometric
Last synced: about 1 month ago
JSON representation
A utility script to install PyTorch Geometric with its dependencies, using the pre-built binaries.
- Host: GitHub
- URL: https://github.com/mberr/pytorch-geometric-installer
- Owner: mberr
- License: mit
- Created: 2021-01-08T10:34:11.000Z (over 4 years ago)
- Default Branch: main
- Last Pushed: 2022-04-30T16:42:24.000Z (about 3 years ago)
- Last Synced: 2025-03-24T12:56:18.026Z (about 2 months ago)
- Topics: pytorch, pytorch-geometric
- Language: Python
- Homepage:
- Size: 12.7 KB
- Stars: 8
- Watchers: 1
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# PyTorch Geometric Install Helper
*Update: PyTorch Geometric now has a similar tool [in their documention](https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html#quick-start).*
Generate pip install commands for installation of [PyTorch Geometric](https://github.com/rusty1s/pytorch_geometric) as described [here](https://github.com/rusty1s/pytorch_geometric#installation), e.g.
```console
$ 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](https://github.com/pytorch/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.