pyg-lib
pyg-lib copied to clipboard
pyg-team
`import pyg_lib` fails due to `AttributeError: module 'pyg_lib' has no attribute 'ops'`
🐛 Describe the bug
import torch_sparse
When I run the code above, it raise an error that pyg_lib has no attribute 'ops'. How shoud I solve this problem?
AttributeError Traceback (most recent call last)
Cell In[48], line 2
1 # import pyg_lib
----> 2 import torch_sparse
File ~/venvs/versions/3.12.4/torch/lib/python3.12/site-packages/torch_sparse/__init__.py:39
31 if t_major != major:
32 raise RuntimeError(
33 f'Detected that PyTorch and torch_sparse were compiled with '
34 f'different CUDA versions. PyTorch has CUDA version '
35 f'{t_major}.{t_minor} and torch_sparse has CUDA version '
36 f'{major}.{minor}. Please reinstall the torch_sparse that '
37 f'matches your PyTorch install.')
---> 39 from .storage import SparseStorage # noqa
40 from .tensor import SparseTensor # noqa
41 from .transpose import t # noqa
File ~/venvs/versions/3.12.4/torch/lib/python3.12/site-packages/torch_sparse/storage.py:7
4 import torch
5 from torch_scatter import scatter_add, segment_csr
----> 7 from torch_sparse.utils import Final, index_sort
9 layouts: Final[List[str]] = ['coo', 'csr', 'csc']
12 def get_layout(layout: Optional[str] = None) -> str:
File ~/venvs/versions/3.12.4/torch/lib/python3.12/site-packages/torch_sparse/utils.py:5
1 from typing import Any, Optional, Tuple
3 import torch
----> 5 import torch_sparse.typing
6 from torch_sparse.typing import pyg_lib
8 try:
File ~/venvs/versions/3.12.4/torch/lib/python3.12/site-packages/torch_sparse/typing.py:4
2 import pyg_lib # noqa
3 WITH_PYG_LIB = True
----> 4 WITH_INDEX_SORT = hasattr(pyg_lib.ops, 'index_sort')
5 except ImportError:
6 pyg_lib = object
AttributeError: module 'pyg_lib' has no attribute 'ops'
Environment
-
pyg-libversion:0.4.0+pt23 - PyTorch version:
2.3.1 - OS:
MacOS - Python version:
3.12.4 - CUDA/cuDNN version:
- How you installed PyTorch and
pyg-lib(conda,pip, source):pip - Any other relevant information:
