mamba icon indicating copy to clipboard operation
mamba copied to clipboard
verified publisher icon state-spaces

Problems with import mamba-ssm

Open yangtianjin opened this issue 2 years ago • 2 comments

import mamba-ssm Traceback (most recent call last): File "", line 1, in File "/mnt/mount/data/ytj/anaconda3/envs/pytorch/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 "/mnt/mount/data/ytj/anaconda3/envs/pytorch/lib/python3.10/site-packages/mamba_ssm/ops/selective_scan_interface.py", line 11, in import selective_scan_cuda ImportError: /mnt/mount/data/ytj/anaconda3/envs/pytorch/lib/python3.10/site-packages/selective_scan_cuda.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN2at4_ops10zeros_like4callERKNS_6TensorEN3c108optionalINS5_10ScalarTypeEEENS6_INS5_6LayoutEEENS6_INS5_6DeviceEEENS6_IbEENS6_INS5_12MemoryFormatEEE

yangtianjin avatar Mar 17 '24 14:03 yangtianjin

same problem. In a nividia container... works well on the previous version of the mamba code.

pengzhangzhi avatar Mar 18 '24 04:03 pengzhangzhi

Downgrade to torch 2.2.2 works for me.

vincent317 avatar Apr 24 '24 19:04 vincent317