search

state-spaces / mamba

Mamba SSM architecture
Apache License 2.0
11.42k stars 928 forks source link

mamba_ssm安装成功，导入失败，无法使用。 #338

Open Lijuming33 opened 1 month ago

Lijuming33 commented 1 month ago

from mamba_ssm import Mamba Traceback (most recent call last): File "", line 1, in File "/home/deng/miniforge-pypy3/envs/mamba/lib/python3.10/site-packages/mamba_ssm/init.py", line 3, in from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn File "/home/deng/miniforge-pypy3/envs/mamba/lib/python3.10/site-packages/mamba_ssm/ops/selective_scan_interface.py", line 16, in import selective_scan_cuda ImportError: /home/deng/miniforge-pypy3/envs/mamba/lib/python3.10/site-packages/selective_scan_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c104cuda9SetDeviceEi

image

wsl2 Ubuntu22.04，gcc>>>11.4 我的cuda版本为11.8，torch=2.3 CUDA11.8为https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run 安装方式为torch安装方式为： pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Lijuming33 commented 1 month ago

cxx11baiTRUE版本也尝试了，一模一样的问题。

catalpaaa commented 3 weeks ago

you should clone the repo down and install it with pip -e, same for conv1d