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.11 and no CUDA support:
CUDA=cpu
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
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 .