search

state-spaces / mamba

Mamba SSM architecture
Apache License 2.0
12.76k stars 1.08k forks source link

import SSM problem #331

Open zhanke199423 opened 4 months ago

zhanke199423 commented 4 months ago

I run the same code on colab successfully yesterday but cannot run it today. And it returns me with error about "import selective_scan_cuda". I dont know why it happens,

File "/content/drive/MyDrive/Beijing_test/gatmamba_simple.py", line 212, in from mamba_ssm import Mamba File "/usr/local/lib/python3.10/dist-packages/mamba_ssm/init.py", line 3, in from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn File "/usr/local/lib/python3.10/dist-packages/mamba_ssm/ops/selective_scan_interface.py", line 16, in import selective_scan_cuda ImportError: /usr/local/lib/python3.10/dist-packages/selective_scan_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN3c104cuda9SetDeviceEi

zixianwang2022 commented 4 months ago

Hi, check out this link: https://github.com/state-spaces/mamba/issues/97#issuecomment-2125852419

tridao commented 4 months ago

v1.2.1 now includes wheels for pytorch 2.3 so it should hopefully fix this issue.