Undefined symbol while importing 'torch-sparse'

[matteotolloso]

Hello everyone, I have the following issue using torch-sparse:

CUDA Version: 12.4
Architecture: aarch64
OS: Ubuntu 22.04.5 LTS

This is what I did:

conda create -n test python=3.11
conda activate test
pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124
pip install torch_geometric
pip install torch-sparse -f https://data.pyg.org/whl/torch-2.5.0+cu124.html

When I do

pip freeze | grep torch

I get:
torch==2.5.1 torch-geometric==2.6.1 torch_sparse==0.6.18 torchaudio==2.5.1 torchvision==0.20.1

The problem is:

import torch_geometric

/vast/user/miniconda3/envs/test/lib/python3.11/site-packages/torch_geometric/typing.py:124: UserWarning: An issue occurred while importing 'torch-sparse'. Disabling its usage. Stacktrace: /vast/user/miniconda3/envs/test/lib/python3.11/site-packages/torch_sparse/_spmm_cpu.so: undefined symbol: _ZN5torch8autograd12VariableInfoC1ERKN2at6TensorEb

Thanks for your help!