ImportError: libcudart.so.12

[SirSykon]

Hello!

I'm trying to run u-mamba and I got an unexpected error. After following the installation, when I execute import mamba_ssm I get error:
File "/opt/conda/envs/umamba/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 "/opt/conda/envs/umamba/lib/python3.10/site-packages/mamba_ssm/ops/selective_scan_interface.py", line 20, in <module> import selective_scan_cuda ImportError: libcudart.so.12: cannot open shared object file: No such file or directory

I'm using pytorch 2.0.1+cu118, so I don't get why the code is looking for a CUDA 12 code. Any clue?

Thanks!

[ZhuMuMu0216]

same question

[fuyuao000]

This looks very likely to be caused by installing the cu12 version of mamba_ssm. Although I don't know exactly how this happened, I suggest downloading a corresponding version of the wheel file from the releases of the official repositories for mamba and causal_conv1d. Additionally, since mamba updates very quickly, there might be differences in syntax and code structure between different versions. I recommend the version combination I am currently using: torch=2.1.1+cu118, mamba_ssm=2.2.1, causal_conv1d=1.3.0.post1, transformer=4.44.2.