From e7cc748de55d8c59204bddf41e667673af550163 Mon Sep 17 00:00:00 2001 From: Mario Geiger Date: Tue, 7 Jun 2022 15:49:53 +0200 Subject: [PATCH] Update INSTALL.md --- INSTALL.md | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/INSTALL.md b/INSTALL.md index 4d4799ae..33a92ca3 100644 --- a/INSTALL.md +++ b/INSTALL.md @@ -8,16 +8,13 @@ e3nn requires PyTorch >=1.8.0 For installation instructions, please see the [PyT ### optional: torch_geometric -First you have to install [pytorch_geometric](https://github.com/rusty1s/pytorch_geometric). For `torch` 1.8.* and no CUDA support: +First you have to install [pytorch_geometric](https://github.com/rusty1s/pytorch_geometric). For `torch` 1.11 and no CUDA support: ```bash -TORCH=1.8.0 CUDA=cpu -pip install --upgrade --force-reinstall torch-scatter -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html -pip install --upgrade --force-reinstall torch-sparse -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html -pip install --upgrade --force-reinstall torch-cluster -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html -pip install --upgrade --force-reinstall torch-spline-conv -f https://pytorch-geometric.com/whl/torch-$TORCH+$CUDA.html +pip install --upgrade --force-reinstall torch-scatter -f https://data.pyg.org/whl/torch-1.11.0+${CUDA}.html +pip install --upgrade --force-reinstall torch-sparse -f https://data.pyg.org/whl/torch-1.11.0+${CUDA}.html pip install torch-geometric ```