Install

Dependencies

PyTorch

e3nn requires PyTorch >=1.8.0 For installation instructions, please see the PyTorch homepage.

optional: torch_geometric

First you have to install pytorch_geometric. For torch 1.8.* and no CUDA support:

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 torch-geometric

See here to get cuda support or newer versions.

e3nn

Stable (PyPI)

$ pip install e3nn

Unstable (Git)

$ git clone https://github.com/e3nn/e3nn.git
$ cd e3nn/
$ pip install .