注意1:mamba-ssm要与casual-conv1d一起安装。
注意2:mamba-ssm与cuda、pytorch版本要对应。需要看你下载的代码的requirements.txt
causal-conv1d与mamba的whl包官网下载:
https://github.com/Dao-AILab/causal-conv1d/releases?page=3
https://github.com/state-spaces/mamba/releases/
mamba要下载abiFALSE的,这个是更稳定的,相比于abiTrue
通过官网下载完对应的xxx.whl文件后,使用如下命令安装
cd 下载好的whl文件的同一目录下
pip3 install xxx.whl(你刚刚下载的文件)
我使用的版本:
cuda 11.8
python3.9
pytorch2.0.1
mamba-ssm1.2.0
causal-conv1d1.2.0