error when installing mamba_ssm

[RICKand-MORTY]

Hi! I downloaded the file mamba_ssm-2.2.4+cu12torch2.5cxx11abiFALSE-cp312-cp312-linux_x86_64.whl and used pip install to installed mamba_ssm. However, when I used mamba, I met the following error:
File "/root/miniconda3/lib/python3.12/site-packages/mamba_ssm/ops/selective_scan_interface.py", line 18, in <module> import selective_scan_cuda ImportError: /root/miniconda3/lib/python3.12/site-packages/selective_scan_cuda.cpython-312-x86_64-linux-gnu.so: undefined symbol: _ZN3c1021throwNullDataPtrErrorEv
How should I solve the problem? Should I reinstall cuda driver?

[Lynnzake]

Similar problem occured to me.
import selective_scan_cuda
ImportError: /home/aiscuser/.conda/envs/mamba/lib/python3.11/site-packages/selective_scan_cuda.cpython-311-x86_64-linux-gnu.so: undefined symbol: _ZN3c1021throwNullDataPtrErrorEv

python=3.11 pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-cuda=12.1

[Lynnzake]

I try code below solved this problem:

git clone https://github.com/state-spaces/mamba.git
pip install . --no-build-isolation

[RICKand-MORTY]

Thanks! I have solved the problem finally with your solution

[sky1502]

Installing torch 2.2.x solved it for me. I guess this symbol has something to do with some change made between torch versions 2.2.x and above