Skip to content

Commit e7cc748

Browse files
authored
Update INSTALL.md
1 parent de7cb22 commit e7cc748

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

INSTALL.md

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,16 +8,13 @@ e3nn requires PyTorch >=1.8.0 For installation instructions, please see the [PyT
88

99
### optional: torch_geometric
1010

11-
First you have to install [pytorch_geometric](https://github.com/rusty1s/pytorch_geometric). For `torch` 1.8.* and no CUDA support:
11+
First you have to install [pytorch_geometric](https://github.com/rusty1s/pytorch_geometric). For `torch` 1.11 and no CUDA support:
1212

1313
```bash
14-
TORCH=1.8.0
1514
CUDA=cpu
1615

17-
pip install --upgrade --force-reinstall torch-scatter -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html
18-
pip install --upgrade --force-reinstall torch-sparse -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html
19-
pip install --upgrade --force-reinstall torch-cluster -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html
20-
pip install --upgrade --force-reinstall torch-spline-conv -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html
16+
pip install --upgrade --force-reinstall torch-scatter -f https://data.pyg.org/whl/torch-1.11.0+${CUDA}.html
17+
pip install --upgrade --force-reinstall torch-sparse -f https://data.pyg.org/whl/torch-1.11.0+${CUDA}.html
2118
pip install torch-geometric
2219
```
2320

0 commit comments

Comments
 (0)