Skip to content

GitLab

"examples/pytorch/han/train_sampling.py" did not exist on "e56bbafd2541199fc61e3cce5d373a5cca0861ea"
Commit fd3a719e authored by rusty1s's avatar rusty1s
Browse files

new pyg version

parent 01a3e346
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
README.md
View file @ fd3a719e
...@@ -36,12 +36,13 @@ class GNN(ScalableGNN): ...@@ -36,12 +36,13 @@ class GNN(ScalableGNN):
## Installation ## Installation
* Install [**PyTorch >= 1.7.0**](https://pytorch.org/get-started/locally/) * Install [**PyTorch >= 1.7.0**](https://pytorch.org/get-started/locally/)
* Install [**PyTorch Geometric**](https://github.com/rusty1s/pytorch_geometric#pytorch-170171) from **master**: * Install [**PyTorch Geometric >= 1.7.0**](https://github.com/rusty1s/pytorch_geometric#installation)
For example, via
``` ```
pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+${CUDA}.html pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+${CUDA}.html
pip install --no-index torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+${CUDA}.html pip install --no-index torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+${CUDA}.html
pip install git+https://github.com/rusty1s/pytorch_geometric.git pip install torch-geometric
``` ```
where `${CUDA}` should be replaced by either `cpu`, `cu92`, `cu101`, `cu102`, or `cu110` depending on your PyTorch installation. where `${CUDA}` should be replaced by either `cpu`, `cu92`, `cu101`, `cu102`, or `cu110` depending on your PyTorch installation.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment