No module named 'torch_cluster.graclus_cuda'

[josauder]

I Install the library using pip3 install torch-cluster.

When using the library using from torch_cluster import knn_graph I get:

Traceback (most recent call last):
  File "/home/jonathan/workspace/pcnn/models/Hybrid.py", line 9, in <module>
    from torch_cluster import knn_graph
  File "/home/jonathan/.local/lib/python3.6/site-packages/torch_cluster/__init__.py", line 1, in <module>
    from .graclus import graclus_cluster
  File "/home/jonathan/.local/lib/python3.6/site-packages/torch_cluster/graclus.py", line 5, in <module>
    import torch_cluster.graclus_cuda
ModuleNotFoundError: No module named 'torch_cluster.graclus_cuda'

How can I fix this?
Thanks in advance.

[rusty1s]

Your CUDA is not setup correctly. Please follow the installation guide.

[divyanshj16]

@josauder were you able to solve this issue?
Pytorch works perfectly with cudatoolkit installed by Anaconda, but this import fails.

[rusty1s]

You need to manually install CUDA to make this package work.

[divyanshj16]

Oh, Alright. Thanks.

[gortali]

I have the same problem, have you managed to solve the issue? my CUDA installation seems fine

[divyanshj16]

Reinstall all the libraries again and build them again.
pip install --no-binary torch-cluster torch-cluster --no-cache-dir
Do this for all libraries, replacing the name of the library at the appropriate place.

[github-actions]

This issue had no activity for 6 months. It will be closed in 2 weeks unless there is some new activity. Is this issue already resolved?