File tree Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Original file line number Diff line number Diff line change @@ -8,16 +8,13 @@ e3nn requires PyTorch >=1.8.0 For installation instructions, please see the [PyT
8
8
9
9
### optional: torch_geometric
10
10
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:
12
12
13
13
``` bash
14
- TORCH=1.8.0
15
14
CUDA=cpu
16
15
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
21
18
pip install torch-geometric
22
19
```
23
20
You can’t perform that action at this time.
0 commit comments