ImportError

[gthd]

I am trying to import mamba_ssm and getting the following error message:

>>> from mamba_ssm import Mamba
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/gtheodorou/.local/lib/python3.10/site-packages/mamba_ssm/__init__.py", line 3, in <module>
    from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
  File "/home/gtheodorou/.local/lib/python3.10/site-packages/mamba_ssm/ops/selective_scan_interface.py", line 18, in <module>
    import selective_scan_cuda
ImportError: /home/gtheodorou/.local/lib/python3.10/site-packages/selective_scan_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c107WarningC1ESt7variantIJNS0_11UserWarningENS0_18DeprecationWarningEEERKNS_14SourceLocationESsb

causal-conv1d Version: 1.5.0.post8 mamba-ssm Version: 2.2.4

nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2024 NVIDIA Corporation Built on Fri_Jun_14_16:34:21_PDT_2024 Cuda compilation tools, release 12.6, V12.6.20 Build cuda_12.6.r12.6/compiler.34431801_0

NVIDIA-SMI 560.35.03 Driver Version: 560.35.03 CUDA Version: 12.6

python -c "import torch; print(torch.version)": 2.6.0+cu126

[supgy]

Hi,have you solved it yet? I encountered the same problem as you~