1、直接使用pip install mamba-ssm,存在的问题(undefined symbol: _ZN3c104cuda9SetDeviceEi in import selective_scan_cuda)
猜测原因:没有安装causal_conv1d
2、解决办法——首先下载whl文件,然后pip install **whl文件
(1)从Releases · Dao-AILab/causal-conv1d · GitHub中选择对应的causal_conv1d
其中cu118表示cuda版本至少大于等于11.8;cxx11abiTrue和cxx11abiFALSE表示关于C++ ABI的标志,cxx11abiFALSE兼容性更好,一般选择它;cp310表示Python 3.10
(2)从Releases · state-spaces/mamba · GitHub中选择对应的mamba-ssm版本
(3)安装指令
pip install causal_conv1d-1.4.0+cu118torch2.3cxx11abiFALSE-cp38-cp38-linux_x86_64.whl # 优先安装它;然后安装mamba-ssm
pip install mamba_ssm-2.2.2+cu118torch2.3cxx11abiFALSE-cp38-cp38-linux_x86_64.whl
备注:我的设备是Ubuntu 20.04,Python 3.8,cuda 11.8