mamba icon indicating copy to clipboard operation
mamba copied to clipboard

mamba_ssm安装成功,导入失败,无法使用。

Open Lijuming33 opened this issue 1 year ago • 4 comments

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 avatar May 31 '24 14:05 Lijuming33

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

Lijuming33 avatar May 31 '24 14:05 Lijuming33

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

catalpaaa avatar Jun 10 '24 06:06 catalpaaa

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

are you solved????

Jon-Zbw avatar Aug 19 '24 08:08 Jon-Zbw

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

are you solved????

yes,我使用conda安装的torch+cuda

Lijuming33 avatar Aug 23 '24 15:08 Lijuming33