Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

ModuleNotFoundError: No module named 'basis_cuda'

See original GitHub issue

Hi, thanks for your excellent work! I am using Ubunt 16.04, cuda 9.2 and pytorch 0.4.1 When importing torch_geometric by: from torch_geometric.nn import GCNConv

I met the following error:

  File "/home/x/PycharmProjects/untitled/foo/simple_gcn.py", line 3, in <module>
    from torch_geometric.nn import GCNConv
  File "/home/x/anaconda3/lib/python3.6/site-packages/torch_geometric-0.3.1-py3.6.egg/torch_geometric/nn/__init__.py", line 1, in <module>
  File "/home/x/anaconda3/lib/python3.6/site-packages/torch_geometric-0.3.1-py3.6.egg/torch_geometric/nn/conv/__init__.py", line 1, in <module>
  File "/home/x/anaconda3/lib/python3.6/site-packages/torch_geometric-0.3.1-py3.6.egg/torch_geometric/nn/conv/spline_conv.py", line 3, in <module>
  File "/home/x/anaconda3/lib/python3.6/site-packages/torch_spline_conv/__init__.py", line 1, in <module>
    from .basis import SplineBasis
  File "/home/x/anaconda3/lib/python3.6/site-packages/torch_spline_conv/basis.py", line 5, in <module>
    import basis_cuda
ModuleNotFoundError: No module named 'basis_cuda'

Currently I can work with my data on CPU, but maybe using GPU may accelerate things up. Could you give me some advice about this?