Open FrancescoCasalegno opened 2 years ago
PyPI seems to have PyTorch-Geometric and its various torch-scatter
... dependencies:
torch-geometric
: https://pypi.org/project/torch-geometrictorch-scatter
: https://pypi.org/project/torch-scattertorch-sparse
: https://pypi.org/project/torch-sparseSo:
torch-scatter
with -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
instead?pip install ...
from PyPI, but it breaks for GPUs?install.sh
for GPU install and usage.It turns out that torch-scatter
, etc. are indeed on PyPI. But only the source tarball is there, so pip install torch-scatter
will take some time to compile the C extensions!
On the other hand, torch-geometric
is quick to install with pip install torch-geometric
even if only the source tarball is on PyPI because torch-geometric
is a pure-Python package.
This could be complicated by the fact that the installation of PyTorch-Geometric is not as trivial as
pip install torch-geometric
.This is particularly true for the
torch-scatter
,torch-sparse
, ... dependencies oftorch-geometric
. For these packages, it seems like wheels are not on PyPI but must be found instead onhttps://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
.Notice that we do need such dependencies for our project: https://github.com/BlueBrain/morphoclass/blob/72f59fad5dca9eb08a5752a21de7577c47f1011b/install.sh#L54-L57